Searched refs:gradient_def (Results 1 – 1 of 1) sorted by relevance
662 GradientDef gradient_def; in RewriteAssociatedFunction() local663 gradient_def.set_function_name(func.name()); in RewriteAssociatedFunction()664 gradient_def.set_gradient_func(rewritten_function_name); in RewriteAssociatedFunction()667 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()669 TF_RETURN_IF_ERROR(fld->ReplaceGradient(gradient_def)); in RewriteAssociatedFunction()