Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.Observation Module Reference

Classes

class  Mila::Dnn::Compute::ComputePassMask
 A set of compute passes, small enough to test on the publication path. More...
struct  Mila::Dnn::Compute::ObservableStage
 A stage a component publishes, and the passes it is published on. More...

Enumerations

enum class  Mila::Dnn::Compute::ComputePass : uint8_t { Forward = 0 , Prefill = 1 , Decode = 2 , Backward = 3 }
 The compute entry point that produced an observed value. More...

Typedefs

using Mila::Dnn::Compute::ActivationObserver
 Receives one published activation.

Functions

constexpr std::string_view Mila::Dnn::Compute::computePassToString (ComputePass pass) noexcept

Files

file  Mila/Src/Dnn/Compute/Observation.ixx
 Vocabulary for activation observation: the compute pass, the stage, the observer.