Mila::Dnn::Router< TDeviceType, TPrecision >::Routing Struct Reference

One call's result. Both views are router-owned and valid until the next forward(). More...

Public Attributes
IndexTensorType &	indices
TensorType &	weights

Detailed Description

template<DeviceType TDeviceType, TensorDataType TPrecision>
struct Mila::Dnn::Router< TDeviceType, TPrecision >::Routing

One call's result. Both views are router-owned and valid until the next forward().

---

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

• Mila/Src/Dnn/Components/MixtureOfExperts/Router.ixx