Searched refs:AddGradientDef (Results 1 – 5 of 5) sorted by relevance
1038 TEST(FunctionLibraryDefinitionTest, AddGradientDef) { in TEST() argument1047 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1050 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1054 Status s = lib_def.AddGradientDef(grad); in TEST()1095 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()
435 Status AddGradientDef(const GradientDef& grad) TF_LOCKS_EXCLUDED(mu_);
1334 Status FunctionLibraryDefinition::AddGradientDef(const GradientDef& grad) { in AddGradientDef() function in tensorflow::FunctionLibraryDefinition1742 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()
729 TF_RETURN_IF_ERROR(flib_def->AddGradientDef(grad_def)); in Convert()
844 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()