Module burn::backend::autodiff::ops

Operation module.

Structs

• ComputePropertyDone
  Operation’s compute property is fixed
• Init
  Operation is initialized
• MemoryBound
  Operation has been tagged as memory bound
• MemoryBoundRetroForward
  Memory bound operation has received its RetroForward
• 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.

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.
• unary_different_backend
  Execute a unary operation during the backward step where the input backend is different from the output backend.