Home
last modified time | relevance | path

Searched defs:slice_offsets (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc100 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), zero_s32); in PadEachPartitionWithHaloExchange() local
Dspmd_partitioner_util.cc492 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in TileToPartialReplicateHaloExchange() local
600 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in PadFromPartialReplicateShape() local
1004 std::vector<HloInstruction*> slice_offsets(base_shape.rank(), zero); in ExchangeHaloAndGetValidData() local
Ddot_handler.cc3218 std::vector<HloInstruction*> slice_offsets(padded_shape.rank()); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
Dspmd_partitioner.cc3339 std::vector<HloInstruction*> slice_offsets(shard_shape.rank(), zero); in HandleSelectAndScatter() local