Searched refs:_make_indexed_slices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | cross_device_ops_test.py | 83 def _make_indexed_slices(values, indices, dense_shape, device): function
93 values = [_make_indexed_slices(values, indices, dense_shape, d)
293 t0 = _make_indexed_slices([[1., 2.]], [1], [5, 2], devices[0])
294 t1 = _make_indexed_slices([[3., 4.], [5., 6.]], [1, 3], [5, 2], devices[1])
302 total_with_dups = _make_indexed_slices(
304 total_without_dups = _make_indexed_slices(
327 t0 = _make_indexed_slices([[1., 2.]], [1], dense_shape, devices[0])
328 t1 = _make_indexed_slices(
|