Home
last modified time | relevance | path

Searched refs:_grad (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/memory_tests/
Dcustom_gradient_memory_test.py43 def _grad(self, test_func, argnums=0): member in RecomputeGradMemoryTest
85 return self._grad(test_func)(a)
87 return def_function.function(self._grad(test_func))(a)
146 test_func = def_function.function(self._grad(test_func), jit_compile=True)
/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py255 def _grad(op, grad): function
329 return _grad
337 def _grad(op, grad): function
369 return _grad
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py1475 def _grad(self, f, argnums=0): member in VariablesGradientTest
1496 f=self._grad(f),
1520 grads_re = self._grad(custom_gradient.recompute_grad(F))(x)
1521 grads = self._grad(F)(x)
1525 grads_re = self._grad(custom_gradient.recompute_grad(f_graph))(x)
1526 grads = self._grad(f_graph)(x)
1536 grads_re = self._grad(custom_gradient.recompute_grad(F))(x1, x2)
1537 grads = self._grad(F)(x1, x2)
1546 grads_re = self._grad(custom_gradient.recompute_grad(f_graph))(x1, x2)
1547 grads = self._grad(f_graph)(x1, x2)
/external/tensorflow/tensorflow/python/eager/
Dfunction_gradients_test.py129 def _grad(f): function
148 f = _grad(f)
149 traced_f = def_function.function(_grad(traced_f))
153 def _grad(f): function
174 f = _grad(f)
Dbackprop_test.py228 def _grad(_): function
234 return x, _grad
1822 def _grad(f): function
1841 f = _grad(f)
Dforwardprop_test.py111 def _grad(f, argnums=0): function
215 f=_grad(f),
/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
Dcond_v2_test.py965 def _grad(f): function
981 f = _grad(f)
/external/tensorflow/tensorflow/core/ir/
DREADME.md85 [tf.recompute\_grad](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/cus…