Searched refs:AddGradientDef (Results 1 – 5 of 5) sorted by relevance
1046 TEST(FunctionLibraryDefinitionTest, AddGradientDef) { in TEST() argument1055 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1058 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1062 Status s = lib_def.AddGradientDef(grad); in TEST()1079 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()
346 Status AddGradientDef(const GradientDef& grad) LOCKS_EXCLUDED(mu_);
1106 Status FunctionLibraryDefinition::AddGradientDef(const GradientDef& grad) { in AddGradientDef() function in tensorflow::FunctionLibraryDefinition1477 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()
282 TF_RETURN_IF_ERROR(library->AddGradientDef(grad_def)); in AddMissingFunctionDef()
667 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()