Searched refs:_grad_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | rev_block_lib.py | 604 def _grad_fn(output_grads, variables=None): function
629 return _grad_fn(output_grads, kwargs["variables"])
632 return _grad_fn(output_grads)
|
| /external/tensorflow/tensorflow/contrib/autograph/examples/benchmarks/ |
| D | cartpole_benchmark.py | 301 self._grad_fn = eager.implicit_gradients(
326 grads_and_vars = self._grad_fn(tf.constant([obs], dtype=tf.float32))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 260 def _grad_fn(func_graph, grads): function
313 lambda: _grad_fn(func_graph, grads), [], {},
|
| D | while_v2.py | 454 lambda *args: _grad_fn(ys, xs, args, body_graph),
475 def _grad_fn(ys, xs, args, func_graph): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 631 return self._grad_fn(op, *doutputs)
633 def _grad_fn(self, op, *doutputs): member in ConcreteFunction
888 return self._grad_fn(call_op, *args)
|