Searched refs:FindGradient (Results 1 – 8 of 8) sorted by relevance
631 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined"); in TEST_F()635 EXPECT_EQ(graph_.flib_def().FindGradient("XTimesTwo"), "Undefined"); in TEST_F()
285 const string gradient_func = lib_def.FindGradient(function_name); in AddFunction()
403 string FindGradient(const string& func) const LOCKS_EXCLUDED(mu_);
1455 string FunctionLibraryDefinition::FindGradient(const string& func) const { in FindGradient() function in tensorflow::FunctionLibraryDefinition1651 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions()1672 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition()
1010 const string grad = flib_def.FindGradient(func.name()); in MakeFunctionBodyForInlining()1438 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass()
658 string original_grad_func = fld->FindGradient(func.name()); in RewriteAssociatedFunction()
775 const string grad = lib_def->FindGradient(func.name()); in Instantiate()
1095 auto grad_func_name = func_lib.FindGradient(std::string(func_name)); in ConvertLibFunction()