Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dtf-gradient-attr.mlir29 // CHECK-NEXT: gradient_func: "foo_grad"
/external/tensorflow/tensorflow/core/framework/
Dfunction.proto108 // A gradient function g (specified by gradient_func) for a function f
125 string gradient_func = 2; // The gradient function's name. field
Ddataset.cc287 const string gradient_func = lib_def.FindGradient(function_name); in AddFunction() local
288 if (!gradient_func.empty()) { in AddFunction()
291 g_def->set_gradient_func(gradient_func); in AddFunction()
Dfunction.cc1204 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()
/external/tensorflow/tensorflow/python/framework/
Dfunction.py1160 if g.gradient_func not in funcs:
1162 (g.gradient_func, str(lib)))
1170 func_to_grad[gdef.function_name] = gdef.gradient_func
1171 grad_to_funcs[gdef.gradient_func].append(gdef.function_name)
Dfunction_def_to_graph.py225 grad_def.gradient_func = f.grad_func_name
Dfunction_test.py1322 gradient.gradient_func = G1.name
1361 gradient1.gradient_func = F2.name
1365 gradient2.gradient_func = F1.name
Dops.py3243 grad_def.gradient_func = f.grad_func_name
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-gradient-def.pbtxt272 gradient_func: "foo_grad"
/external/tensorflow/tensorflow/c/
Dc_test_util.cc438 grads.emplace_back(grad.function_name(), grad.gradient_func()); in GetGradDefs()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor_test.cc2648 EXPECT_EQ(gdef.library().gradient()[0].gradient_func(), "FooGrad_dc60abc8"); in TEST_F()