Searched refs:CleanUpGradTensor (Results 1 – 3 of 3) sorted by relevance
66 CleanUpGradTensor(grad); in TryApplyGrad()90 virtual void CleanUpGradTensor(GradientTensorType* tensor) = 0;
129 void CleanUpGradTensor(const Tensor* tensor) override { in CleanUpGradTensor() function
404 void CleanUpGradTensor(std::tuple<const Tensor*, const Tensor*, in CleanUpGradTensor() function