Searched refs:ReplaceGradient (Results 1 – 3 of 3) sorted by relevance
431 Status ReplaceGradient(const GradientDef& grad) TF_LOCKS_EXCLUDED(mu_);
1436 Status FunctionLibraryDefinition::ReplaceGradient(const GradientDef& grad) { in ReplaceGradient() function in tensorflow::FunctionLibraryDefinition
668 TF_RETURN_IF_ERROR(fld->ReplaceGradient(gradient_def)); in RewriteAssociatedFunction()