Expand description
Defines runtime-executable compute kernels.
A kernel is a pure compute primitive that reads one or more input Tensors and writes
results into a pre-allocated output Tensor. Kernels do not allocate output storage and
are expected to validate basic argument invariants (arity, shape, rank) before any compute.
To allocate Tensors, see the crate::graph::Graph API.
Structs§
- AddKernel
- Elementwise addition kernel.
- MatMul
Kernel - Naive 2-D matrix multiplication kernel.
- Relu
Kernel - Rectified Linear Unit (
ReLU) activation kernel.
Enums§
- Kernel
Error - Kernel-level errors raised during argument validation or execution.
Traits§
- Kernel
- A runtime-executable compute primitive.