Searched defs:grad_fn (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | grad_testutil.cc | 26 ops::GradFunc grad_fn; in CallGradFunction() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2_test.py | 224 def grad_fn(dy): function
246 def grad_fn(dy): function
|
| D | gradients_util.py | 322 def _MaybeCompile(scope, op, func, grad_fn): argument
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 375 ops::GradFunc grad_fn; in IsPrimitiveOpWithNoGrad() local
384 ops::GradFunc grad_fn; in CallGradFunction() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 228 def grad_fn(*args, **kwds): function
301 def grad_fn(*args, **kwds): function
|
| D | backprop_test.py | 1603 def grad_fn(x): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cholesky_op_test.py | 357 def _BenchmarkGrad(grad_fn, name, device): argument
|
| D | while_v2_test.py | 117 …def grad_fn(dy, variables=None): # pylint: disable=invalid-name, unused-argument, redefined-outer… function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy_test.py | 134 def grad_fn(x): function
|
| D | parameter_server_strategy_test.py | 463 def grad_fn(x): function
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 49 def get_filtered_grad_fn(grad_fn): argument
|