Searched refs:ReshapeToTileDimension (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util_test.cc | 164 ReshapeToTileDimension(sharding, /*dim=*/0, /*dims=*/{0, 1}); in TEST()
171 ReshapeToTileDimension(sharding, /*dim=*/1, /*dims=*/{0, 1}); in TEST()
179 ReshapeToTileDimension(sharding, /*dim=*/0, /*dims=*/{0, 1, 2}); in TEST()
189 ReshapeToTileDimension(sharding, /*dim=*/1, /*dims=*/{0, 1, 2}); in TEST()
198 ReshapeToTileDimension(sharding, /*dim=*/2, /*dims=*/{0, 1, 2}); in TEST()
209 HloSharding result = ReshapeToTileDimension(sharding, /*dim=*/2, in TEST()
|
| D | hlo_sharding_util.h | 103 HloSharding ReshapeToTileDimension(const HloSharding& sharding, int64 dim,
|
| D | hlo_sharding_util.cc | 433 HloSharding ReshapeToTileDimension(const HloSharding& sharding, int64 dim, in ReshapeToTileDimension() function
|