Searched refs:kMaxAuxiliaryParameterCount (Results 1 – 2 of 2) sorted by relevance
67 static constexpr int kMaxAuxiliaryParameterCount = 3; variable
184 *support = auxiliary_parameter_count + 1 <= kMaxAuxiliaryParameterCount in GetGradientAccumulationSupport()326 if (state_variables->size() > kMaxAuxiliaryParameterCount + 1) { in GetOptimizationAlgorithmStateVariables()