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.cc98 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), zero_s32); in PadEachPartitionWithHaloExchange() local
Dspmd_partitioner_util.cc407 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in TileToPartialReplicateHaloExchange() local
516 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in PadFromPartialReplicateShape() local
966 std::vector<HloInstruction*> slice_offsets(base_shape.rank(), zero); in ExchangeHaloAndGetValidData() local
Ddot_handler.cc4180 std::vector<HloInstruction*> slice_offsets; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() member
4233 HloInstruction* dus) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4338 HloInstruction* last_iter_result) -> StatusOr<HloInstruction*> { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
Dspmd_partitioner.cc4148 std::vector<HloInstruction*> slice_offsets(shard_shape.rank(), zero); in HandleSelectAndScatter() local