Home
last modified time | relevance | path

Searched refs:out_grads (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py369 def _SymGrad(op, out_grads): argument
371 f_in = [x for x in op.inputs] + out_grads
658 out_grads = _AggregatedGrads(grads, op, gradient_uid, loop_state,
670 has_out_grads = any(isinstance(g, ops.Tensor) or g for g in out_grads)
711 for i, out_grad in enumerate(out_grads):
720 out_grads[i] = loop_state.ZerosLike(op, i)
722 out_grads[i] = control_flow_ops.ZerosLikeOutsideLoop(op, i)
731 lambda: grad_fn(op, *out_grads))
736 lambda: _SymGrad(op, out_grads))
747 _LogOpGradients(op, out_grads, in_grads)
[all …]
Darray_grad.py106 out_grads = []
116 out_grads = array_ops.split(grad, sizes, non_neg_concat_dim)
147 out_grads = array_ops.split(grad, sizes, non_neg_concat_dim)
151 out_grads.append(array_ops.slice(grad, begin, size))
180 out_grads.append(ops.IndexedSlices(new_values, grad.indices, size))
201 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
206 return (out_grads + [None]
207 if end_value_index <= dim_index else [None] + out_grads)
/external/tensorflow/tensorflow/compiler/tests/
Dlrn_ops_test.py116 out_grads = constant_op.constant(out_grads_vals, shape=shape)
118 expected = gen_nn_ops.lrn_grad(out_grads, in_image, out_image,
121 actual = gen_nn_ops.lrn_grad(out_grads, in_image, out_image,
/external/tensorflow/tensorflow/contrib/batching/python/ops/
Dbatch_ops.py32 def _BatchGrad(op, *out_grads): # pylint: disable=invalid-name argument
38 out_grads[i],
/external/tensorflow/tensorflow/core/profiler/
DREADME.md202 …gradients_impl.py:554:gradients:out_grads[i] = co..., cpu: 164.71ms, accelerator: 16.21ms, total: …
206 …gradients_impl.py:515:gradients:out_grads = _Aggr..., cpu: 108.69ms, accelerator: 51.92ms, total: …
207 …gradients_impl.py:846:_AggregatedGrads:out_grads[i] = _M..., cpu: 107.99ms, accelerator: 50.05ms, …
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py114 out_grads): argument
133 return grad_fn(mock_op, *out_grads)
Dpywrap_tfe_src.cc1639 new PyBackwardFunction([backward_function](PyObject* out_grads) { argument
1640 return PyObject_CallObject(backward_function, out_grads);