Searched refs:gradient_def (Results 1 – 1 of 1) sorted by relevance
661 GradientDef gradient_def; in RewriteAssociatedFunction() local662 gradient_def.set_function_name(func.name()); in RewriteAssociatedFunction()663 gradient_def.set_gradient_func(rewritten_function_name); in RewriteAssociatedFunction()666 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()668 TF_RETURN_IF_ERROR(fld->ReplaceGradient(gradient_def)); in RewriteAssociatedFunction()