Home
last modified time | relevance | path

Searched refs:GetGradientAccumulationSupport (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_embedding_optimization_parameters_utils.cc196 Status GetGradientAccumulationSupport(const OptimizationParameters& params, in GetGradientAccumulationSupport() function
210 TF_RETURN_IF_ERROR(GetGradientAccumulationSupport(params, &support)); in UseGradientAccumulation()
H A Dtpu_embedding_optimization_parameters_utils.h56 Status GetGradientAccumulationSupport(const OptimizationParameters &params,