Searched refs:IndexedSlices (Results 1 – 25 of 97) sorted by relevance
1234
| /external/tensorflow/tensorflow/python/ops/ |
| D | clip_ops.py | 148 t.values if isinstance(t, ops.IndexedSlices) else t, name="t")
163 if isinstance(t, ops.IndexedSlices):
164 return ops.IndexedSlices(values_clip, t.indices, t.dense_shape)
199 t.values if isinstance(t, ops.IndexedSlices) else t,
282 t.values if isinstance(t, ops.IndexedSlices) else t,
297 ops.IndexedSlices(c_v, t.indices, t.dense_shape)
298 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 | gradients_util.py | 99 ops.register_tensor_conversion_function(ops.IndexedSlices,
263 if isinstance(grad_y, ops.IndexedSlices):
265 ops.IndexedSlices(
785 if isinstance(out_grad, (ops.Tensor, ops.IndexedSlices)):
875 assert isinstance(grad, ops.IndexedSlices)
879 assert isinstance(grad.values, ops.IndexedSlices)
881 return ops.IndexedSlices(g.values, array_ops.gather(
999 if isinstance(out_grad, (ops.Tensor, ops.IndexedSlices)):
1004 isinstance(g, (ops.Tensor, ops.IndexedSlices))
1070 concat_grad = ops.IndexedSlices(
|
| D | clip_ops_test.py | 49 indixed_slices = ops.IndexedSlices(values, indices, shape)
52 self.assertIsInstance(clipped, ops.IndexedSlices)
|
| D | array_grad.py | 152 elif isinstance(grad, ops.IndexedSlices):
180 out_grads.append(ops.IndexedSlices(new_values, grad.indices, size))
201 out_grads.append(ops.IndexedSlices(new_values, new_indices, size))
411 return [ops.IndexedSlices(values, indices, params_shape), None]
442 return [ops.IndexedSlices(values, indices, params_shape), None, None]
485 ref_grad = ops.IndexedSlices(grad, array_ops.squeeze(indices, axis=-1),
585 if isinstance(grad, ops.IndexedSlices):
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | training.py | 282 if isinstance(grad, ops.IndexedSlices):
310 if isinstance(grad, ops.IndexedSlices):
312 grad = ops.IndexedSlices(tmp, grad.indices, grad.dense_shape)
355 if isinstance(grad, ops.IndexedSlices):
358 grad = ops.IndexedSlices(tmp, grad.indices, grad.dense_shape)
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | optimizers.py | 275 if isinstance(gradient, ops.IndexedSlices):
400 elif isinstance(grad, ops.IndexedSlices):
402 ops.IndexedSlices(grad.values * factor, grad.indices,
420 if isinstance(gradient, ops.IndexedSlices):
437 if isinstance(grad, ops.IndexedSlices):
439 grad = ops.IndexedSlices(grad_values, grad.indices, grad.dense_shape)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | addsign.py | 144 sign_g = ops.IndexedSlices(
146 sign_gm = ops.IndexedSlices(
154 multiplier = ops.IndexedSlices(
157 final_update = ops.IndexedSlices(
|
| D | powersign.py | 147 sign_g = ops.IndexedSlices(
149 sign_gm = ops.IndexedSlices(
158 multiplier = ops.IndexedSlices(
161 final_update = ops.IndexedSlices(
|
| D | reg_adagrad_optimizer_test.py | 122 grads0 = ops.IndexedSlices(
125 grads1 = ops.IndexedSlices(
151 grad_repeated_index = ops.IndexedSlices(
154 grad_aggregated = ops.IndexedSlices(
205 grads0 = ops.IndexedSlices(
312 grads0 = ops.IndexedSlices(
315 grads1 = ops.IndexedSlices(
|
| D | variable_clipping_optimizer.py | 120 assert isinstance(grad, ops.IndexedSlices)
133 delta = ops.IndexedSlices(
|
| D | adamax_test.py | 93 grads0 = ops.IndexedSlices(
97 grads1 = ops.IndexedSlices(
150 grad_repeated_index = ops.IndexedSlices(
155 grad_aggregated = ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 647 if any(isinstance(v, ops.IndexedSlices) for v in values):
654 if isinstance(value, ops.IndexedSlices):
656 return ops.IndexedSlices(
664 if isinstance(value, ops.IndexedSlices):
668 result = ops.IndexedSlices(copied_values, copied_indices, copied_shape)
676 if isinstance(value, ops.IndexedSlices):
687 if isinstance(value, ops.IndexedSlices):
690 return all([isinstance(v, ops.IndexedSlices) for v in value.values])
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | adagrad_test.py | 119 grads0 = ops.IndexedSlices(
124 grads1 = ops.IndexedSlices(
152 grad_repeated_index = ops.IndexedSlices(
157 grad_aggregated = ops.IndexedSlices(
209 grads0 = ops.IndexedSlices(
|
| D | adam_test.py | 74 grads0 = ops.IndexedSlices(
78 grads1 = ops.IndexedSlices(
130 grad_repeated_index = ops.IndexedSlices(
135 grad_aggregated = ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| D | learning.py | 293 if isinstance(grad, ops.IndexedSlices):
295 grad = ops.IndexedSlices(tmp, grad.indices, grad.dense_shape)
335 if isinstance(grad, ops.IndexedSlices):
337 grad = ops.IndexedSlices(tmp, grad.indices, grad.dense_shape)
356 if isinstance(grad, 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.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.ops.IndexedSlices\'>"
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_utils_test.py | 57 self.assertIsInstance(result, ops.IndexedSlices)
75 self.assertIsInstance(result, ops.IndexedSlices)
135 self.assertIsInstance(result, ops.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(
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_allgather_test.py | 88 to_gather.append(tf.IndexedSlices(concat_vals, concat_ind))
94 correct_gather = tf.IndexedSlices(correct_gather_values,
|
| D | __init__.py | 150 if isinstance(tensor, tf.IndexedSlices):
159 return tf.IndexedSlices(new_values, indices,
|
| /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(
670 ops.IndexedSlices(
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | ftrl_test.py | 238 grads0 = ops.IndexedSlices(
241 grads1 = ops.IndexedSlices(
313 grads0 = ops.IndexedSlices(
316 grads1 = ops.IndexedSlices(
|
1234