Searched refs:gradient_func (Results 1 – 11 of 11) sorted by relevance
29 // CHECK-NEXT: gradient_func: "foo_grad"
108 // A gradient function g (specified by gradient_func) for a function f125 string gradient_func = 2; // The gradient function's name. field
287 const string gradient_func = lib_def.FindGradient(function_name); in AddFunction() local288 if (!gradient_func.empty()) { in AddFunction()291 g_def->set_gradient_func(gradient_func); in AddFunction()
1204 func_grad_[grad.function_name()] = grad.gradient_func(); in FunctionLibraryDefinition()1342 if (*entry != grad.gradient_func()) { in AddGradientDefHelper()1344 "Cannot assign gradient function '", grad.gradient_func(), "' to '", in AddGradientDefHelper()1351 *entry = grad.gradient_func(); in AddGradientDefHelper()
1160 if g.gradient_func not in funcs:1162 (g.gradient_func, str(lib)))1170 func_to_grad[gdef.function_name] = gdef.gradient_func1171 grad_to_funcs[gdef.gradient_func].append(gdef.function_name)
225 grad_def.gradient_func = f.grad_func_name
1322 gradient.gradient_func = G1.name1361 gradient1.gradient_func = F2.name1365 gradient2.gradient_func = F1.name
3243 grad_def.gradient_func = f.grad_func_name
272 gradient_func: "foo_grad"
438 grads.emplace_back(grad.function_name(), grad.gradient_func()); in GetGradDefs()
2648 EXPECT_EQ(gdef.library().gradient()[0].gradient_func(), "FooGrad_dc60abc8"); in TEST_F()