| /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)
217 t.values if isinstance(t, ops.IndexedSlices) else t, name="t")
232 if isinstance(t, ops.IndexedSlices):
233 return ops.IndexedSlices(values_clip, t.indices, t.dense_shape)
269 t.values if isinstance(t, ops.IndexedSlices) else t,
359 t.values if isinstance(t, ops.IndexedSlices) else t,
374 ops.IndexedSlices(c_v, t.indices, t.dense_shape)
375 if isinstance(t, ops.IndexedSlices)
|
| D | while_v2_indexed_slices_rewriter.py | 70 if not isinstance(output, ops.IndexedSlices): continue
145 input_slices = ops.IndexedSlices(
178 assert isinstance(grad_output_slices, ops.IndexedSlices)
210 return ops.IndexedSlices(values=values, indices=indices, dense_shape=shape)
244 return ops.IndexedSlices(values=values, indices=indices,
|
| D | clip_ops_test.py | 59 indexed_slices = ops.IndexedSlices(values, indices, shape)
62 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))
555 if not isinstance(indexed_slices, ops.IndexedSlices):
582 return [ops.IndexedSlices(values, indices, params_shape), None]
670 params_grad = ops.IndexedSlices(values, indices, params_shape)
715 ref_grad = ops.IndexedSlices(grad, array_ops.squeeze(indices, axis=-1),
728 ref_grad = ops.IndexedSlices(grad, array_ops.squeeze(indices, axis=-1),
842 if isinstance(grad, ops.IndexedSlices):
|
| D | control_flow_grad.py | 166 if not isinstance(grad, (ops.IndexedSlices, sparse_tensor.SparseTensor)):
225 elif isinstance(grad, ops.IndexedSlices):
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 546 return ops.IndexedSlices(
559 return ops.IndexedSlices(
576 if any(isinstance(v, ops.IndexedSlices) for v in values):
583 if isinstance(value, ops.IndexedSlices):
585 return ops.IndexedSlices(
593 if isinstance(value, ops.IndexedSlices):
597 result = ops.IndexedSlices(copied_values, copied_indices, copied_shape)
604 if isinstance(value, ops.IndexedSlices):
607 return all(isinstance(v, ops.IndexedSlices) for v in value.values)
|
| D | vars_test.py | 466 value = indexed_slices.IndexedSlices(
490 value = indexed_slices.IndexedSlices(
511 value = indexed_slices.IndexedSlices(
532 value = indexed_slices.IndexedSlices(
555 value = indexed_slices.IndexedSlices(
582 value = indexed_slices.IndexedSlices(
609 value = indexed_slices.IndexedSlices(
632 value = indexed_slices.IndexedSlices(
1157 indexed_slices.IndexedSlices(
1159 indexed_slices.IndexedSlices(
[all …]
|
| D | cross_device_ops_test.py | 59 IndexedSlices = indexed_slices.IndexedSlices variable
85 IndexedSlices(
203 elif isinstance(value, IndexedSlices):
396 expect = IndexedSlices(
399 expect = IndexedSlices(
404 expect = IndexedSlices(
420 expect = IndexedSlices(
534 IndexedSlices(
541 IndexedSlices(
545 IndexedSlices(
[all …]
|
| D | cross_device_utils_test.py | 58 self.assertIsInstance(result, ops.IndexedSlices)
76 self.assertIsInstance(result, ops.IndexedSlices)
110 self.assertIsInstance(result, ops.IndexedSlices)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | indexed_slices.py | 61 class IndexedSlices(internal.NativeObject, composite_tensor.CompositeTensor): class
166 return IndexedSlices(-self.values, self.indices, self.dense_shape)
211 value_type = property(lambda self: IndexedSlices)
268 return IndexedSlices(*tensor_list)
455 IndexedSlices, _indexed_slices_to_tensor)
|
| D | python_tensor_converter_test.py | 175 x = indexed_slices.IndexedSlices(
187 x = indexed_slices.IndexedSlices(
199 x = indexed_slices.IndexedSlices(
|
| 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/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/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/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(
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 587 assert isinstance(grad, ops.IndexedSlices)
591 assert isinstance(grad.values, ops.IndexedSlices)
593 return ops.IndexedSlices(g.values, array_ops.gather(grad.indices,
617 concat_grad = ops.IndexedSlices(
642 assert all(isinstance(g, (ops.Tensor, ops.IndexedSlices))
651 elif isinstance(grad, 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 | 580 ops.IndexedSlices(
593 ops.IndexedSlices(
606 ops.IndexedSlices(
616 ops.IndexedSlices(
629 ops.IndexedSlices(
639 ops.IndexedSlices(
652 ops.IndexedSlices(
665 ops.IndexedSlices(
678 ops.IndexedSlices(
1046 slices = ops.IndexedSlices(indices=[1], values=[2])
[all …]
|
| D | clip_ops_test.py | 184 indexed_slices = ops.IndexedSlices(values, indices, shape)
188 self.assertIsInstance(clipped, ops.IndexedSlices)
378 x1 = ops.IndexedSlices(
398 slices = ops.IndexedSlices(
|
| D | sparsemask_op_test.py | 44 t = ops.IndexedSlices(values_tensor, indices_tensor)
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adam_test.py | 127 grads0 = ops.IndexedSlices(
131 grads1 = ops.IndexedSlices(
181 grad_repeated_index = ops.IndexedSlices(
186 grad_aggregated = ops.IndexedSlices(
317 grad_repeated_index = ops.IndexedSlices(
320 grad_aggregated = ops.IndexedSlices(grads0_np, indices,
579 grads0 = ops.IndexedSlices(
583 grads1 = ops.IndexedSlices(
633 grad_repeated_index = ops.IndexedSlices(
638 grad_aggregated = ops.IndexedSlices(
[all …]
|
| D | adagrad_test.py | 320 grads0 = ops.IndexedSlices(
324 grads1 = ops.IndexedSlices(
362 grads0 = ops.IndexedSlices(
398 grad_repeated_index = ops.IndexedSlices(
401 grad_aggregated = ops.IndexedSlices(
458 grads0 = ops.IndexedSlices(
|
| D | adamax_test.py | 97 grads0 = ops.IndexedSlices(
101 grads1 = ops.IndexedSlices(
151 grad_repeated_index = ops.IndexedSlices(
156 grad_aggregated = ops.IndexedSlices(
|
| D | gradient_descent_test.py | 213 grads0 = ops.IndexedSlices(
216 grads1 = ops.IndexedSlices(
236 grads0 = ops.IndexedSlices(
239 grads1 = ops.IndexedSlices(
415 y_feed = ops.IndexedSlices(x_feed, constant_op.constant([0, 1]),
530 grads0 = ops.IndexedSlices(
533 grads1 = ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 151 if isinstance(grad, ops.IndexedSlices):
153 return ops.IndexedSlices(grad_vals, grad.indices, grad.dense_shape)
|