Searched defs:grad_fn (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | grad_testutil.cc | 26 ops::GradFunc grad_fn; in CallGradFunction() local
|
| D | nn_grad.cc | 416 BatchNormGradFn grad_fn, in BaseFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker_v2_test.py | 220 def grad_fn(dy): function
242 def grad_fn(dy): function
|
| D | gradients_util.py | 311 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 | 236 def grad_fn(*args, **kwds): function
309 def grad_fn(*args, **kwds): function
|
| D | backprop_test.py | 1660 def grad_fn(x): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | cholesky_op_test.py | 352 def _BenchmarkGrad(grad_fn, name, device): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 459 def grad_fn(x): function
|
| D | collective_all_reduce_strategy_test.py | 158 def grad_fn(x): function
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 44 def get_filtered_grad_fn(grad_fn): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| D | while_v2_test.py | 116 …def grad_fn(dy, variables=None): # pylint: disable=invalid-name, unused-argument, redefined-outer… function
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_test.py | 2077 def grad_fn(*args): function
|