Searched refs:MakePartitionOffsets (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 105 std::vector<HloInstruction*> MakePartitionOffsets(
|
| D | gather_scatter_handler.cc | 57 auto operand_offsets = MakePartitionOffsets( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
375 std::vector<HloInstruction*> operand_offsets = MakePartitionOffsets( in PartitionIndexParallelDimensions()
|
| D | spmd_partitioner_util.cc | 204 std::vector<HloInstruction*> MakePartitionOffsets( in MakePartitionOffsets() function
247 return MakePartitionOffsets(table_shape, sharding, partition_id, b); in MakeTiledPartitionOrdinals()
1730 MakePartitionOffsets(replicated->shape(), group_level_sharding, group_id, in PerGroupSliceFromReplicated()
|
| D | spmd_partitioner.cc | 473 MakePartitionOffsets(shape, target, state_.partition_id, state_.b), in ReshardNoCache()
517 auto offsets = MakePartitionOffsets(base_shape_, sharding, in PadWithValue()
888 auto offsets = MakePartitionOffsets(padded_target_shape, sharding(), in ReplicatePartial()
1033 auto offsets = MakePartitionOffsets(padded_base_shape, temp_target_sharding, in ReshardFromPartialReplicateWithDynamicSlice()
1035 auto old_offsets = MakePartitionOffsets(padded_base_shape, sharding(), in ReshardFromPartialReplicateWithDynamicSlice()
|
| D | dot_handler.cc | 1031 auto ccw_offsets = MakePartitionOffsets( in PartitionBaseCase()
1036 auto cw_offsets = MakePartitionOffsets( in PartitionBaseCase()
1115 auto offsets = MakePartitionOffsets( in PartitionBaseCase()
|