Searched refs:_make_indexed_slices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | cross_device_ops_test.py | 99 def _make_indexed_slices(values, indices, dense_shape, device): function
109 values = [_make_indexed_slices(values, indices, dense_shape, d)
240 t0 = _make_indexed_slices([[1., 2.]], [1], dense_shape, devices[0])
241 t1 = _make_indexed_slices([[3., 4.], [5., 6.]], [1, 3], dense_shape,
370 t0 = _make_indexed_slices([[1., 2.]], [1], [5, 2], devices[0])
371 t1 = _make_indexed_slices([[3., 4.], [5., 6.]], [1, 3], [5, 2], devices[1])
378 total_with_dups = _make_indexed_slices(
380 total_without_dups = _make_indexed_slices(
657 _make_indexed_slices(
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 210 t0 = _make_indexed_slices(
528 t0 = _make_indexed_slices(
658 def _make_indexed_slices(values, indices, dense_shape): function
|
| D | strategy_common_test.py | 166 def _make_indexed_slices(values, indices, dense_shape): function
|