Searched refs:FindGradient (Results 1 – 8 of 8) sorted by relevance
275 const string grad_name = lookup_library->FindGradient(func_name); in AddMissingFunctionDef()276 if (!grad_name.empty() && library->FindGradient(func_name).empty()) { in AddMissingFunctionDef()
665 string original_grad_func = fld->FindGradient(func.name()); in RewriteAssociatedFunction()
631 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined"); in TEST_F()635 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined"); in TEST_F()
267 const string gradient_func = ctx->flib_def().FindGradient(function_name); in AddFunction()
379 string FindGradient(const string& func) const LOCKS_EXCLUDED(mu_);
1257 string FunctionLibraryDefinition::FindGradient(const string& func) const { in FindGradient() function in tensorflow::FunctionLibraryDefinition1438 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions()1471 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition()
2129 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass()
712 const string grad = lib_def->FindGradient(func.name()); in Instantiate()