| /external/tensorflow/tensorflow/python/ops/ |
| D | clip_ops.py | 108 t.values if isinstance(t, ops.IndexedSlices) else t, name="t")
119 if isinstance(t, ops.IndexedSlices):
120 t_max = ops.IndexedSlices(t_max, t.indices, t.dense_shape)
195 t.values if isinstance(t, ops.IndexedSlices) else t, name="t")
210 if isinstance(t, ops.IndexedSlices):
211 return ops.IndexedSlices(values_clip, t.indices, t.dense_shape)
246 t.values if isinstance(t, ops.IndexedSlices) else t,
337 t.values if isinstance(t, ops.IndexedSlices) else t,
352 ops.IndexedSlices(c_v, t.indices, t.dense_shape)
353 if isinstance(t, ops.IndexedSlices)
|
| D | while_v2_indexed_slices_rewriter.py | 70 if not isinstance(output, ops.IndexedSlices): continue
136 input_slices = ops.IndexedSlices(
169 assert isinstance(grad_output_slices, ops.IndexedSlices)
201 return ops.IndexedSlices(values=values, indices=indices, dense_shape=shape)
235 return ops.IndexedSlices(values=values, indices=indices,
|
| D | clip_ops_test.py | 49 indixed_slices = ops.IndexedSlices(values, indices, shape)
52 self.assertIsInstance(clipped, ops.IndexedSlices)
|
| D | array_grad.py | 158 elif isinstance(grad, ops.IndexedSlices):
186 out_grads.append(ops.IndexedSlices(new_values, grad.indices, size))
207 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
522 if not isinstance(indexed_slices, ops.IndexedSlices):
555 return [ops.IndexedSlices(values, indices, params_shape), None]
642 params_grad = ops.IndexedSlices(values, indices, params_shape)
687 ref_grad = ops.IndexedSlices(grad, array_ops.squeeze(indices, axis=-1),
700 ref_grad = ops.IndexedSlices(grad, array_ops.squeeze(indices, axis=-1),
814 if isinstance(grad, ops.IndexedSlices):
|
| D | control_flow_grad.py | 166 if not isinstance(grad, (ops.IndexedSlices, sparse_tensor.SparseTensor)):
223 elif isinstance(grad, ops.IndexedSlices):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | indexed_slices.py | 63 class IndexedSlices(_TensorLike, composite_tensor.CompositeTensor): class
156 return IndexedSlices(-self.values, self.indices, self.dense_shape)
201 value_type = property(lambda self: IndexedSlices)
258 return IndexedSlices(*tensor_list)
441 IndexedSlices, _indexed_slices_to_tensor)
|
| D | composite_tensor_test.py | 381 ('IndexedSlicesNoDenseShape', lambda: ops.IndexedSlices(
383 ('IndexedSlicesInt32DenseShape', lambda: ops.IndexedSlices(
386 ('IndexedSlicesInt64DenseShape', lambda: ops.IndexedSlices(
397 ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 688 if any(isinstance(v, ops.IndexedSlices) for v in values):
695 if isinstance(value, ops.IndexedSlices):
697 return ops.IndexedSlices(
705 if isinstance(value, ops.IndexedSlices):
709 result = ops.IndexedSlices(copied_values, copied_indices, copied_shape)
717 if isinstance(value, ops.IndexedSlices):
728 if isinstance(value, ops.IndexedSlices):
731 return all(isinstance(v, ops.IndexedSlices) for v in value.values)
|
| D | cross_device_utils_test.py | 57 self.assertIsInstance(result, ops.IndexedSlices)
75 self.assertIsInstance(result, ops.IndexedSlices)
134 self.assertIsInstance(result, ops.IndexedSlices)
|
| D | cross_device_ops_test.py | 91 tensor = ops.IndexedSlices(
112 self.assertIsInstance(left, ops.IndexedSlices)
113 self.assertIsInstance(right, ops.IndexedSlices)
132 if isinstance(left, ops.IndexedSlices):
520 if isinstance(left, ops.IndexedSlices):
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-indexed-slices.pbtxt | 1 path: "tensorflow.IndexedSlices"
3 is_instance: "<class \'tensorflow.python.framework.indexed_slices.IndexedSlices\'>"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-indexed-slices.pbtxt | 1 path: "tensorflow.IndexedSlices"
3 is_instance: "<class \'tensorflow.python.framework.indexed_slices.IndexedSlices\'>"
|
| /external/tensorflow/tensorflow/python/training/ |
| D | adagrad_test.py | 153 grads0 = ops.IndexedSlices(
158 grads1 = ops.IndexedSlices(
187 grad_repeated_index = ops.IndexedSlices(
192 grad_aggregated = ops.IndexedSlices(
246 grads0 = ops.IndexedSlices(
|
| D | adam_test.py | 74 grads0 = ops.IndexedSlices(
78 grads1 = ops.IndexedSlices(
135 grad_repeated_index = ops.IndexedSlices(
140 grad_aggregated = ops.IndexedSlices(
|
| D | ftrl_test.py | 236 grads0 = ops.IndexedSlices(
239 grads1 = ops.IndexedSlices(
311 grads0 = ops.IndexedSlices(
314 grads1 = ops.IndexedSlices(
|
| D | optimizer.py | 126 assert isinstance(g, ops.IndexedSlices), ("Gradient ", g, " is neither a "
165 if isinstance(g, ops.IndexedSlices):
585 if not isinstance(g, (ops.Tensor, ops.IndexedSlices)):
686 if not isinstance(g, (ops.Tensor, ops.IndexedSlices)):
1034 gradient_no_duplicate_indices = ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_conditional_accumulator_test.py | 43 return ops.IndexedSlices(
123 ops.IndexedSlices(
189 grad_indexed_slices = ops.IndexedSlices(
210 grad_indexed_slices = ops.IndexedSlices(
237 grad_indexed_slices = ops.IndexedSlices(
253 grad_indexed_slices = ops.IndexedSlices(
685 ops.IndexedSlices(
|
| D | resource_variable_ops_test.py | 547 v.scatter_add(ops.IndexedSlices(indices=[1], values=[2.5])))
555 v.scatter_sub(ops.IndexedSlices(indices=[1], values=[1.5])))
563 v.scatter_max(ops.IndexedSlices(indices=[1], values=[5.0])))
569 v.scatter_max(ops.IndexedSlices(indices=[1], values=[2.0])))
577 v.scatter_min(ops.IndexedSlices(indices=[1], values=[5.0])))
583 v.scatter_min(ops.IndexedSlices(indices=[1], values=[2.0])))
591 v.scatter_mul(ops.IndexedSlices(indices=[1], values=[3.0])))
599 v.scatter_div(ops.IndexedSlices(indices=[1], values=[2.0])))
607 v.scatter_update(ops.IndexedSlices(indices=[1], values=[3.0])))
974 slices = ops.IndexedSlices(indices=[1], values=[2])
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 563 assert isinstance(grad, ops.IndexedSlices)
567 assert isinstance(grad.values, ops.IndexedSlices)
569 return ops.IndexedSlices(g.values, array_ops.gather(grad.indices,
594 concat_grad = ops.IndexedSlices(
619 assert all(isinstance(g, (ops.Tensor, ops.IndexedSlices))
631 if isinstance(grad, ops.IndexedSlices):
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 147 if isinstance(grad, ops.IndexedSlices):
149 return ops.IndexedSlices(grad_vals, grad.indices, grad.dense_shape)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adagrad_test.py | 316 grads0 = ops.IndexedSlices(
320 grads1 = ops.IndexedSlices(
356 grads0 = ops.IndexedSlices(
391 grad_repeated_index = ops.IndexedSlices(
394 grad_aggregated = ops.IndexedSlices(
451 grads0 = ops.IndexedSlices(
|
| D | ftrl_test.py | 238 grads0 = ops.IndexedSlices(
241 grads1 = ops.IndexedSlices(
313 grads0 = ops.IndexedSlices(
316 grads1 = ops.IndexedSlices(
|
| D | gradient_descent_test.py | 210 grads0 = ops.IndexedSlices(
213 grads1 = ops.IndexedSlices(
232 grads0 = ops.IndexedSlices(
235 grads1 = ops.IndexedSlices(
415 y_feed = ops.IndexedSlices(x_feed, constant_op.constant([0, 1]),
528 grads0 = ops.IndexedSlices(
531 grads1 = ops.IndexedSlices(
|
| D | adam_test.py | 127 grads0 = ops.IndexedSlices(
131 grads1 = ops.IndexedSlices(
180 grad_repeated_index = ops.IndexedSlices(
185 grad_aggregated = ops.IndexedSlices(
320 grad_repeated_index = ops.IndexedSlices(
323 grad_aggregated = ops.IndexedSlices(grads0_np, indices,
|
| D | adamax_test.py | 96 grads0 = ops.IndexedSlices(
100 grads1 = ops.IndexedSlices(
153 grad_repeated_index = ops.IndexedSlices(
158 grad_aggregated = ops.IndexedSlices(
|