Home
last modified time | relevance | path

Searched refs:gradient_func (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddataset.cc267 const string gradient_func = ctx->flib_def().FindGradient(function_name); in AddFunction() local
268 if (!gradient_func.empty()) { in AddFunction()
271 g_def->set_gradient_func(gradient_func); in AddFunction()
Dfunction.proto88 // A gradient function g (specified by gradient_func) for a function f
105 string gradient_func = 2; // The gradient function's name. field
Dfunction.cc1048 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()
/external/tensorflow/tensorflow/python/framework/
Dfunction.py1098 if g.gradient_func not in funcs:
1100 (g.gradient_func, str(lib)))
1108 func_to_grad[gdef.function_name] = gdef.gradient_func
1109 grad_to_funcs[gdef.gradient_func].append(gdef.function_name)
Dfunction_test.py1321 gradient.gradient_func = G1.name
1360 gradient1.gradient_func = F2.name
1364 gradient2.gradient_func = F1.name
Dops.py3254 grad_def.gradient_func = f.grad_func_name
/external/tensorflow/tensorflow/c/
Dc_test_util.cc438 grads.emplace_back(grad.function_name(), grad.gradient_func()); in GetGradDefs()
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor_test.cc2644 EXPECT_EQ(gdef.library().gradient()[0].gradient_func(), "FooGrad_dc60abc8"); in TEST_F()