Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::ObservableStage Struct Referenceexport

A stage a component publishes, and the passes it is published on. More...

Public Attributes

std::string name
ComputePassMask passes

Detailed Description

A stage a component publishes, and the passes it is published on.

The stage set differs by pass: an attention block's prefill produces score intermediates its decode never materializes.


The documentation for this struct was generated from the following file: