Searched refs:gradient_func (Results 1 – 8 of 8) sorted by relevance
267 const string gradient_func = ctx->flib_def().FindGradient(function_name); in AddFunction() local268 if (!gradient_func.empty()) { in AddFunction()271 g_def->set_gradient_func(gradient_func); in AddFunction()
88 // A gradient function g (specified by gradient_func) for a function f105 string gradient_func = 2; // The gradient function's name. field
1048 func_grad_[grad.function_name()] = grad.gradient_func(); in FunctionLibraryDefinition()1117 if (*entry != grad.gradient_func()) { in AddGradientDefHelper()1119 "Cannot assign gradient function '", grad.gradient_func(), "' to '", in AddGradientDefHelper()1126 *entry = grad.gradient_func(); in AddGradientDefHelper()
1098 if g.gradient_func not in funcs:1100 (g.gradient_func, str(lib)))1108 func_to_grad[gdef.function_name] = gdef.gradient_func1109 grad_to_funcs[gdef.gradient_func].append(gdef.function_name)
1321 gradient.gradient_func = G1.name1360 gradient1.gradient_func = F2.name1364 gradient2.gradient_func = F1.name
3254 grad_def.gradient_func = f.grad_func_name
438 grads.emplace_back(grad.function_name(), grad.gradient_func()); in GetGradDefs()
2644 EXPECT_EQ(gdef.library().gradient()[0].gradient_func(), "FooGrad_dc60abc8"); in TEST_F()