Searched refs:GetBaseAuxiliaryParameterCount (Results 1 – 2 of 2) sorted by relevance
108 Status GetBaseAuxiliaryParameterCount(const OptimizationParameters& params, in GetBaseAuxiliaryParameterCount() function183 GetBaseAuxiliaryParameterCount(params, &auxiliary_parameter_count)); in GetGradientAccumulationSupport()300 TF_RETURN_IF_ERROR(GetBaseAuxiliaryParameterCount(params, &num_slots)); in GetOptimizationAlgorithmStateVariables()
51 Status GetBaseAuxiliaryParameterCount(const OptimizationParameters ¶ms,