Expand description
Burn intermediate representation.
Structs§
- Adaptive
AvgPool1d Backward OpIr - Adaptive
AvgPool1d OpIr - Adaptive
AvgPool2d Backward OpIr - Adaptive
AvgPool2d OpIr - AvgPool1d
Backward OpIr - AvgPool1d
OpIr - AvgPool2d
Backward OpIr - AvgPool2d
OpIr - Binary
OpIr - Cast
OpIr - CatOpIr
- Clamp
OpIr - Conv1d
OpIr - Conv1d
Options Ir - Conv2d
OpIr - Conv2d
Options Ir - Conv3d
OpIr - Conv3d
Options Ir - Conv
Transpose1d OpIr - Conv
Transpose1d Options Ir - Conv
Transpose2d OpIr - Conv
Transpose2d Options Ir - Conv
Transpose3d OpIr - Conv
Transpose3d Options Ir - Creation
OpIr - Creation operation intermediate representation. As opposed to InitOperationIr, creation operations are lazy initialized.
- Cross
OpIr - Custom
OpIr - Custom operation in fusion stream, declaring its inputs and outputs.
- Deform
Conv2d Backward OpIr - Deform
Conv2d OpIr - Deformable
Conv2d Options Ir - Dequantize
OpIr - DimOpIr
- IR for operations that operate along a dimension without reducing it.
Unlike
ReduceDimOpIr, the output shape is the same as the input shape. - Embedding
Backward OpIr - Embedding
OpIr - Flip
OpIr - Flip operation intermediate representation.
- Full
OpIr - Full operation intermediate representation.
- Gather
OpIr - Grid
Sample2d OpIr - Grid
Sample Options Ir - Handle
Container - Keep all tensor handles in one place and ensure that all resources are used optimally.
- Init
Operation Ir - Declares a tensor has been initialized.
- Interpolate
Backward OpIr - Interpolate
OpIr - Interpolate
Options Ir - Mask
Fill OpIr - Mask
Where OpIr - Matmul
OpIr - MaxPool1d
OpIr - MaxPool1d
With Indices Backward OpIr - MaxPool1d
With Indices OpIr - MaxPool2d
OpIr - MaxPool2d
With Indices Backward OpIr - MaxPool2d
With Indices OpIr - Permute
OpIr - Permute operation intermediate representation.
- Quantization
Parameters Ir - Quantization parameters intermediate representation.
- Quantize
OpIr - Random
OpIr - Reduce
DimOp Ir - Reduce
DimWith Indices OpIr - Reduce
OpIr - Repeat
DimOp Ir - Scalar
OpIr - Scatter
OpIr - Select
Assign OpIr - Select
OpIr - Shape
OpIr - Shape operation intermediate representation.
- Slice
Assign OpIr - Slice
OpIr - Swap
Dims OpIr - Swap dim operation intermediate representation.
- Tensor
Handle - A tensor representation containing a reference to a tensor resource with a given shape.
- Tensor
Id - The tensor unique identifier.
- Tensor
Ir - A tensor definition represents a snapshot of a tensor when it was used.
- Unary
OpIr - Unfold
OpIr - Unfold operation intermediate representation.
Enums§
- Base
Operation Ir - Basic operations that can be done on any tensor type.
- Bool
Operation Ir - Operation intermediate representation specific to a bool tensor.
- Float
Operation Ir - Operation intermediate representation specific to a float tensor.
- Grid
Sample Padding Mode Ir - Handle
- Backend tensor handle wrapper tracking their creation state
- Handle
Kind - Handle which points to a backend tensor primitive kind.
- IntOperation
Ir - Operation intermediate representation specific to an int tensor.
- Interpolate
Mode Ir - IrError
- Module
Operation Ir - Operation intermediate representation specific to module.
- Numeric
Operation Ir - Numeric operations on int and float tensors.
- Operation
Ir - Describe all tensor operations possible.
- Scalar
Ir - A scalar representation.
- Tensor
Status - The status of the current tensor.
Traits§
- Backend
Ir - Backend extension trait that allows an existing backend to use the Burn tensor intermediate representation for compilation purpose or other…
- Operation
Output - Extension trait to extract outputs when registering an operation.