Searched refs:TileOffsetForDevice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_test.cc | 105 EXPECT_EQ(sharding.TileOffsetForDevice(shape, 0), in TEST_F()
107 EXPECT_EQ(sharding.TileOffsetForDevice(shape, 3), in TEST_F()
109 EXPECT_EQ(sharding.TileOffsetForDevice(shape, 2), in TEST_F()
111 EXPECT_EQ(sharding.TileOffsetForDevice(shape, 1), in TEST_F()
|
| D | hlo_sharding.h | 142 std::vector<int64> TileOffsetForDevice(const Shape& shape,
|
| D | hlo_sharding.cc | 163 std::vector<int64> HloSharding::TileOffsetForDevice(const Shape& shape, in TileOffsetForDevice() function in xla::HloSharding
|