Searched refs:AddGradientDef (Results 1 – 6 of 6) sorted by relevance
1101 TEST(FunctionLibraryDefinitionTest, AddGradientDef) { in TEST() argument1110 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1113 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1117 Status s = lib_def.AddGradientDef(grad); in TEST()1158 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()
437 Status AddGradientDef(const GradientDef& grad) TF_LOCKS_EXCLUDED(mu_);
1374 Status FunctionLibraryDefinition::AddGradientDef(const GradientDef& grad) { in AddGradientDef() function in tensorflow::FunctionLibraryDefinition1782 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()
637 TF_RETURN_IF_ERROR(library.AddGradientDef(*gradient)); in ConvertToFunctionDef()
714 TF_RETURN_IF_ERROR(flib_def->AddGradientDef(grad_def)); in Convert()
835 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()