Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::AMPConfig Member List

This is the complete list of members for Mila::Dnn::Compute::AMPConfig, including all inherited members.

AMPConfig(OpPrecision precision=OpPrecision::Auto)Mila::Dnn::Compute::AMPConfiginline
getComputeType() constMila::Dnn::Compute::AMPConfiginline
getCublasComputeType() constMila::Dnn::Compute::AMPConfiginline
OpPrecision enum nameMila::Dnn::Compute::AMPConfig
precision_Mila::Dnn::Compute::AMPConfigprivate
useLossScaling() constMila::Dnn::Compute::AMPConfiginline
useReducedPrecision() constMila::Dnn::Compute::AMPConfiginline
useTensorCores() constMila::Dnn::Compute::AMPConfiginline