|
Mila
Deep Neural Network Library
|
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. | |