| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 232 grad_func=None, argument
273 self._grad_func = grad_func
336 def set_grad_func(self, grad_func): argument
339 assert isinstance(grad_func, _DefinedFunction)
340 self._grad_func = grad_func
594 grad_func=None, argument
620 assert grad_func is None or isinstance(grad_func, _OverloadedFunction)
621 self._grad_func = grad_func
1082 def _from_definition(fdef, grad_func=None): argument
1107 result = _DefinedFunction(func, argnames, input_types, func_name, grad_func,
[all …]
|
| D | function_test.py | 266 @function.Defun(dtype, dtype, grad_func=XentLossGrad)
296 @function.Defun(dtype, grad_func=Grad)
1130 @function.Defun(dtypes.float32, grad_func=Bar)
1170 def expectFunctionsEqual(self, func, grad_func=None, new_func=None): argument
1177 new_func = function._from_definition(fdef, grad_func=grad_func)
1199 @function.Defun(dtypes.float32, grad_func=G)
1203 self.expectFunctionsEqual(F, grad_func=G)
1250 @function.Defun(dtypes.float32, grad_func=G1)
1254 @function.Defun(dtypes.float32, grad_func=G1)
1259 @function.Defun(dtypes.float32, grad_func=G2)
[all …]
|
| D | importer_test.py | 1168 @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 | 755 auto grad_func = in ConvertLibFunction() local
759 ConvertLibFunction(configs, tf_dialect, grad_func, flib)); in ConvertLibFunction()
762 grad.set_gradient_func(grad_func.getName().str()); in ConvertLibFunction()
|
| D | import_model.cc | 1099 auto grad_func = module_.lookupSymbol<mlir::FuncOp>(mlir_grad_func_name); in ConvertLibFunction() local
1100 auto gradient_attr = builder_.getSymbolRefAttr(grad_func); in ConvertLibFunction()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_test.py | 493 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
496 f = self._GetFunc(grad_func=grad_func)
516 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
521 grad_func=grad_func, python_grad_func=self._PythonGradient)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients_test.py | 370 for grad_func in [gradients.jacobian, gradients.batch_jacobian]:
374 self.assertIsNone(grad_func(y, x, use_pfor=use_pfor))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1438 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local
1439 const bool no_specialize = !grad_func.empty() || in RunFunctionOptimizerPass()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 297 grad_func=_example_noninlined_funcdef_grad,
|