Searched refs:AddGradientDef (Results 1 – 5 of 5) sorted by relevance
1034 TEST(FunctionLibraryDefinitionTest, AddGradientDef) { in TEST() argument1043 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1046 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()1050 Status s = lib_def.AddGradientDef(grad); in TEST()1091 TF_EXPECT_OK(lib_def.AddGradientDef(grad)); in TEST()
419 Status AddGradientDef(const GradientDef& grad) TF_LOCKS_EXCLUDED(mu_);
1331 Status FunctionLibraryDefinition::AddGradientDef(const GradientDef& grad) { in AddGradientDef() function in tensorflow::FunctionLibraryDefinition1720 const Status added_grad = reachable_flib.AddGradientDef(grad); in ReachableFunctionLibraryDefinition()
736 TF_RETURN_IF_ERROR(flib_def->AddGradientDef(grad_def)); in Convert()
666 TF_RETURN_IF_ERROR(fld->AddGradientDef(gradient_def)); in RewriteAssociatedFunction()