| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_op_test.py | 204 out_grad = [[0.1, 0.2], [0.3, 0.4], [0.5, 0.6, 0.7]],
211 out_grad = [],
219 out_grad = [[[0.1, 0.2], [0.3, 0.4]],
227 out_grad = [[[0.1, 0.2]], [[0.3]], [[0.4, 0.5]]],
234 out_grad = [[[0.1, 0.2]], [[0.3]], [[0.4, 0.5]]],
241 out_grad = [],
249 out_grad = [[], [], [[1, 2]], [[3, 4], [5, 6]], [[7, 7]]],
258 out_grad, argument
282 out_grad = ragged_factory_ops.constant(
283 out_grad, dtype=dtypes.float32, ragged_rank=out_ragged_rank)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 660 for i, out_grad in enumerate(out_grads):
661 if (not isinstance(out_grad, ops.Tensor) and not out_grad) and (
737 for out_grad in out_grads:
738 if isinstance(out_grad, (ops.Tensor, ops.IndexedSlices)):
740 if out_grad and isinstance(out_grad, collections_abc.Sequence):
741 if any(g is not None for g in out_grad):
964 for i, out_grad in enumerate(out_grads):
966 if isinstance(out_grad, (ops.Tensor, ops.IndexedSlices)):
970 if (isinstance(out_grad, collections_abc.Sequence) and not all(
972 for g in out_grad
[all …]
|
| D | sparse_grad.py | 100 def _SparseTensorDenseAddGrad(op, out_grad): argument
103 return (None, array_ops.gather_nd(out_grad, sp_indices), None, out_grad)
107 def _SparseReduceSumGrad(op, out_grad): argument
112 out_grad_reshaped = array_ops.reshape(out_grad, output_shape_kept_dims)
|
| D | gradients_test.py | 1154 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1156 grads = gradients.gradients(out, [x, variables[0]], grad_ys=out_grad)
1205 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1207 grads = gradients.gradients(out, [x, variables[0]], grad_ys=out_grad)
1243 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1244 del out_grad
1267 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1268 del out_grad
1358 def Grad(out_grad, variables=None): # pylint: disable=redefined-outer-name argument
1360 grads = gradients.gradients(out, [x, variables[0]], grad_ys=out_grad)
[all …]
|
| D | control_flow_ops_test.py | 1090 out_grad = 3.
1091 actual_grads = tape.gradient(case_out, inputs, output_gradients=out_grad)
1094 expected_grads[used_branch_idx] = out_grad * 2. * used_branch_idx
1126 out_grad = 3.
1127 actual_grads = tape.gradient(case_out, inputs, output_gradients=out_grad)
1137 tape.gradient(y, inputs[used_bi], output_gradients=out_grad)))
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2380 Value out_grad = op.grad(); in matchAndRewrite() local
2382 out_grad.getType().template dyn_cast<RankedTensorType>(); in matchAndRewrite()
2401 out_grad, orig_input_shape, ksize, strides, op, zero, rewriter); in matchAndRewrite()
|