Home
last modified time | relevance | path

Searched refs:grad_func_name (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dfunction.cc1438 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctions() local
1439 if (!grad_func_name.empty()) add_to_func_queue(grad_func_name); in ReachableFunctions()
1471 const string grad_func_name = flib.FindGradient(func_name); in ReachableFunctionLibraryDefinition() local
1472 if (!grad_func_name.empty()) { in ReachableFunctionLibraryDefinition()
1475 grad.set_gradient_func(grad_func_name); in ReachableFunctionLibraryDefinition()
/external/tensorflow/tensorflow/python/framework/
Dfunction_test.py1192 self.assertEqual(func.grad_func_name, new_func.grad_func_name)
1228 self.assertEqual(Foo.grad_func_name, new_func.grad_func_name)
Dops.py3251 if f.grad_func_name:
3254 grad_def.gradient_func = f.grad_func_name
3368 if (function.grad_func_name is not None) and (function.python_grad_func is
Dfunction.py315 def grad_func_name(self): member in _DefinedFunction
/external/tensorflow/tensorflow/python/eager/
Dfunction.py356 self.grad_func_name = None