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.py307 def _SymGrad(op, out_grads): argument
309 f_in = [x for x in op.inputs] + out_grads
593 out_grads = _AggregatedGrads(grads, op, gradient_uid, loop_state,
605 has_out_grads = any(isinstance(g, ops.Tensor) or g for g in out_grads)
660 for i, out_grad in enumerate(out_grads):
670 out_grads[i] = loop_state.ZerosLikeV1WhileLoop(op, i)
675 out_grads[i] = control_flow_state.ZerosLike(op, i)
684 lambda: grad_fn(op, *out_grads))
689 lambda: _SymGrad(op, out_grads))
700 _LogOpGradients(op, out_grads, in_grads)
[all …]
Darray_grad.py112 out_grads = []
122 out_grads = array_ops.split(grad, sizes, non_neg_concat_dim)
153 out_grads = array_ops.split(grad, sizes, non_neg_concat_dim)
157 out_grads.append(array_ops.slice(grad, begin, size))
186 out_grads.append(ops.IndexedSlices(new_values, grad.indices, size))
207 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
212 return (out_grads + [None] if end_value_index <= dim_index else [None] +
213 out_grads)
/external/tensorflow/tensorflow/python/framework/experimental/
Dtape.py45 out_grads = self._c_tape.ComputeGradient(ctx, flat_targets, flat_sources,
47 return nest.pack_sequence_as(sources, 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/python/eager/
Dbackprop.py130 out_grads, skip_input_indices, forward_pass_name_scope): argument
159 return grad_fn(mock_op, *out_grads)
161 return grad_fn(mock_op, *out_grads)
Dpywrap_tfe_src.cc2565 [backward_function](PyObject* out_grads,
2567 return PyObject_CallObject(backward_function, out_grads);
2619 [backward_function](PyObject* out_grads,
2621 return PyObject_CallObject(backward_function, out_grads);
2670 [backward_function](PyObject* out_grads,
2672 return PyObject_CallObject(backward_function, out_grads);
/external/tensorflow/tensorflow/c/eager/
Dtape.h334 const std::vector<Gradient*>& in_grads, absl::Span<Gradient*> out_grads);
928 const std::vector<Gradient*>& in_grads, absl::Span<Gradient*> out_grads) { in ForwardpropFromTape() argument
1018 out_grads); in ForwardpropFromTape()