Home
last modified time | relevance | path

Searched refs:expand_tile_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h369 const std::vector<int64>& expand_tile_dims,
Dspmd_partitioner_util.cc505 const std::vector<int64>& expand_tile_dims, in PadFromPartialReplicateShape() argument
525 for (auto dim : expand_tile_dims) { in PadFromPartialReplicateShape()
Dspmd_partitioner.cc994 std::vector<int64> expand_tile_dims; in ReshardFromPartialReplicateWithDynamicSlice() local
1002 expand_tile_dims.push_back(dim); in ReshardFromPartialReplicateWithDynamicSlice()
1017 hlo_, base_shape_, sharding(), temp_target_sharding, expand_tile_dims, in ReshardFromPartialReplicateWithDynamicSlice()