| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 234 grad_func=None, argument
275 self._grad_func = grad_func
338 def set_grad_func(self, grad_func): argument
341 assert isinstance(grad_func, _DefinedFunction)
342 self._grad_func = grad_func
592 grad_func=None, argument
618 assert grad_func is None or isinstance(grad_func, _OverloadedFunction)
619 self._grad_func = grad_func
1096 def _from_definition(fdef, grad_func=None): argument
1121 result = _DefinedFunction(func, argnames, input_types, func_name, grad_func,
[all …]
|
| D | function_test.py | 267 @function.Defun(dtype, dtype, grad_func=XentLossGrad)
297 @function.Defun(dtype, grad_func=Grad)
1133 @function.Defun(dtypes.float32, grad_func=Bar)
1179 def expectFunctionsEqual(self, func, grad_func=None, new_func=None): argument
1186 new_func = function._from_definition(fdef, grad_func=grad_func)
1210 @function.Defun(dtypes.float32, grad_func=G)
1214 self.expectFunctionsEqual(F, grad_func=G)
1263 @function.Defun(dtypes.float32, grad_func=G1)
1267 @function.Defun(dtypes.float32, grad_func=G1)
1272 @function.Defun(dtypes.float32, grad_func=G2)
[all …]
|
| D | importer_test.py | 1155 @function.Defun(dtype, dtype, grad_func=Grad)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 1322 TF_Function* grad_func; in TEST_F() local
1323 DefineFunction("MyGrad", &grad_func); in TEST_F()
1326 TF_GraphCopyFunction(host_graph_, func_, grad_func, s_); in TEST_F()
1342 TF_GraphCopyFunction(host_graph_, func_, grad_func, s_); in TEST_F()
1349 TF_DeleteFunction(grad_func); in TEST_F()
1366 TF_Function* grad_func; in TEST_F() local
1369 DefineFunction("MyGrad", &grad_func); in TEST_F()
1372 TF_GraphCopyFunction(host_graph_, func1, grad_func, s_); in TEST_F()
1374 TF_GraphCopyFunction(host_graph_, func2, grad_func, s_); in TEST_F()
1389 TF_DeleteFunction(grad_func); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 652 auto grad_func = in ConvertLibFunction() local
656 ConvertLibFunction(configs, tf_dialect, grad_func, flib)); in ConvertLibFunction()
659 grad.set_gradient_func(grad_func.getName().str()); in ConvertLibFunction()
|
| D | import_model.cc | 1279 auto grad_func = module_.lookupSymbol<mlir::FuncOp>(mlir_grad_func_name); in ConvertLibFunction() local
1280 auto gradient_attr = builder_.getSymbolRefAttr(grad_func); in ConvertLibFunction()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_test.py | 550 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
553 f = self._GetFunc(grad_func=grad_func)
573 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
578 grad_func=grad_func, python_grad_func=self._PythonGradient)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 369 for grad_func in [gradients.jacobian, gradients.batch_jacobian]:
373 self.assertIsNone(grad_func(y, x, use_pfor=use_pfor))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1472 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local
1474 !grad_func.empty() || ctx.IsFeedNode(node.name()) || in RunFunctionOptimizerPass()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 321 grad_func=_example_noninlined_funcdef_grad,
|