Home
last modified time | relevance | path

Searched refs:CleanUpGradTensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtyped_conditional_accumulator_base.h66 CleanUpGradTensor(grad); in TryApplyGrad()
90 virtual void CleanUpGradTensor(GradientTensorType* tensor) = 0;
H A Dconditional_accumulator.h126 void CleanUpGradTensor(const Tensor* tensor) override { in CleanUpGradTensor() function
H A Dsparse_conditional_accumulator.h379 void CleanUpGradTensor(std::tuple<const Tensor*, const Tensor*, in CleanUpGradTensor() function