Searched refs:grad_func (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/framework/ |
D | function.py | 210 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 …]
|
D | function_test.py | 266 @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 …]
|
D | importer_test.py | 1168 @function.Defun(dtype, dtype, grad_func=Grad)
|
/external/tensorflow/tensorflow/contrib/opt/python/training/ |
D | external_optimizer.py | 374 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/ |
D | c_api_function_test.cc | 1284 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/ |
D | gradients_test.py | 457 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)
|
D | nn_impl.py | 360 grad_func=_swish_grad,
|
/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 | 2129 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() local 2131 if (specialize_func && grad_func.empty() && specialization_worthy) { in RunFunctionOptimizerPass()
|