Home
last modified time | relevance | path

Searched refs:grad_func (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dfunction.py210 grad_func=None, argument
251 self._grad_func = grad_func
308 def set_grad_func(self, grad_func): argument
311 assert isinstance(grad_func, _DefinedFunction)
312 self._grad_func = grad_func
550 grad_func=None, argument
576 assert grad_func is None or isinstance(grad_func, _OverloadedFunction)
577 self._grad_func = grad_func
1034 def _from_definition(fdef, grad_func=None): argument
1059 result = _DefinedFunction(func, argnames, input_types, func_name, grad_func,
[all …]
Dfunction_test.py266 @function.Defun(dtype, dtype, grad_func=XentLossGrad)
296 @function.Defun(dtype, grad_func=Grad)
1142 @function.Defun(dtypes.float32, grad_func=Bar)
1182 def expectFunctionsEqual(self, func, grad_func=None, new_func=None): argument
1189 new_func = function._from_definition(fdef, grad_func=grad_func)
1211 @function.Defun(dtypes.float32, grad_func=G)
1215 self.expectFunctionsEqual(F, grad_func=G)
1262 @function.Defun(dtypes.float32, grad_func=G1)
1266 @function.Defun(dtypes.float32, grad_func=G1)
1271 @function.Defun(dtypes.float32, grad_func=G2)
[all …]
Dimporter_test.py1168 @function.Defun(dtype, dtype, grad_func=Grad)
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dexternal_optimizer.py374 for func, grad_func in zip(equality_funcs, equality_grad_funcs):
375 constraints.append({'type': 'eq', 'fun': func, 'jac': grad_func})
376 for func, grad_func in zip(inequality_funcs, inequality_grad_funcs):
377 constraints.append({'type': 'ineq', 'fun': func, 'jac': grad_func})
/external/tensorflow/tensorflow/c/
Dc_api_function_test.cc1284 TF_Function* grad_func; in TEST_F() local
1285 DefineFunction("MyGrad", &grad_func); in TEST_F()
1288 TF_GraphCopyFunction(host_graph_, func_, grad_func, s_); in TEST_F()
1304 TF_GraphCopyFunction(host_graph_, func_, grad_func, s_); in TEST_F()
1311 TF_DeleteFunction(grad_func); in TEST_F()
1328 TF_Function* grad_func; in TEST_F() local
1331 DefineFunction("MyGrad", &grad_func); in TEST_F()
1334 TF_GraphCopyFunction(host_graph_, func1, grad_func, s_); in TEST_F()
1336 TF_GraphCopyFunction(host_graph_, func2, grad_func, s_); in TEST_F()
1351 TF_DeleteFunction(grad_func); in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py457 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
460 f = self._GetFunc(grad_func=grad_func)
480 grad_func = framework_function.Defun(dtypes.float32, dtypes.float32,
485 grad_func=grad_func, python_grad_func=self._PythonGradient)
Dnn_impl.py360 grad_func=_swish_grad,
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dgradients_test.py370 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/
Dfunction_optimizer.cc2129 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local
2131 if (specialize_func && grad_func.empty() && specialization_worthy) { in RunFunctionOptimizerPass()