Module ops
Expand description
Operation module.
Structs§
- Compute
Property Done - Operation’s compute property is fixed
- Init
- Operation is initialized
- Memory
Bound - Operation has been tagged as memory bound
- Memory
Bound Retro Forward - Memory bound operation has received its RetroForward
- Node
Guard - Keeps an input node available while a dependent operation is being prepared.
- Ops
- Operation containing its parent nodes, its own node and the backward step state.
- OpsPrep
- Operation in preparation.
- Tracked
- Tracked operation tag.
- UnTracked
- Untracked operation tag.
Enums§
- OpsKind
- Enum used before finishing tracked and untracked operations.
Traits§
- Backward
- Trait for all operations.
- Differentiable
Transfer - Transfers plain floating-point primitives between two backends for a recorded operation.
Functions§
- binary
- Execute a binary operation during the backward step.
- broadcast_
shape - Make sure the grad tensor has the given shape.
- unary
- Execute a unary operation during the backward step.