Searched refs:_testCollectiveGather (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 332 def _testCollectiveGather(self, t0, t1, expected, set_graph_key): member in CollectiveOpTest
353 self._testCollectiveGather([0, 1, 2, 3, 4, 5, 6, 7],
358 self._testCollectiveGather([[0, 1, 2, 3], [4, 5, 6, 7]],
363 self._testCollectiveGather([[[0, 1], [2, 3]], [[4, 5], [6, 7]]],
|