Mila
Deep Neural Network Library
|
This is the complete list of members for Mila::Dnn::Compute::AMPConfig, including all inherited members.
AMPConfig(OpPrecision precision=OpPrecision::Auto) | Mila::Dnn::Compute::AMPConfig | inline |
getComputeType() const | Mila::Dnn::Compute::AMPConfig | inline |
getCublasComputeType() const | Mila::Dnn::Compute::AMPConfig | inline |
OpPrecision enum name | Mila::Dnn::Compute::AMPConfig | |
precision_ | Mila::Dnn::Compute::AMPConfig | private |
useLossScaling() const | Mila::Dnn::Compute::AMPConfig | inline |
useReducedPrecision() const | Mila::Dnn::Compute::AMPConfig | inline |
useTensorCores() const | Mila::Dnn::Compute::AMPConfig | inline |