Searched refs:PartialTile (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 190 *to_merge = HloSharding::PartialTile(new_tile, merged_metadata); in MergeSharding()
293 ? HloSharding::PartialTile(tile_assignment, sharding.metadata()) in TransposeSharding()
407 ? HloSharding::PartialTile(new_tile_assignment, in ReshapeSharding()
428 ? HloSharding::PartialTile(new_tile_assignment, in ReverseSharding()
531 ? HloSharding::PartialTile(new_tile_assignment, in GatherOutputSharding()
582 ? HloSharding::PartialTile(new_tile_assignment, in GatherIndexSharding()
667 ? HloSharding::PartialTile(new_tile_assignment, in ScatterIndexSharding()
700 ? HloSharding::PartialTile(new_tile_assignment, in ScatterDataSharding()
839 ? HloSharding::PartialTile(tile_assignment, in PassthroughOperandToGatherOutputOrScatterUpdate()
886 ? HloSharding::PartialTile(tile_assignment, in PassthroughGatherOutputOrScatterUpdateToOperand()
[all …]
|
| D | hlo_sharding.h | 70 static HloSharding PartialTile(
78 static HloSharding PartialTile(
|
| D | hlo_sharding.cc | 47 HloSharding HloSharding::PartialTile( in PartialTile() function in xla::HloSharding
64 return PartialTile(new_tile_assignment, metadata); in PartialTile()
67 HloSharding HloSharding::PartialTile( in PartialTile() function in xla::HloSharding
524 ? PartialTile(tile_assignment, metadata) in FromProto()
|
| D | sharding_propagation.cc | 446 ? HloSharding::PartialTile( in InferGatherParallelShardingFromOperands()
769 ? HloSharding::PartialTile(new_tile_assignment, in InferShardingFromOperands()
1337 ? HloSharding::PartialTile(tile_assignment, in GetShardingFromUser()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.cc | 382 ? HloSharding::PartialTile(reshape_tile_assignment) in PartialReplicateReshardCompatibleSharding()
404 ? HloSharding::PartialTile(transpose_tile_assignment) in PartialReplicateReshardCompatibleSharding()
1430 ? HloSharding::PartialTile(grouped_tiling) in GroupShardingOnDims()
1478 return partial_sharding ? HloSharding::PartialTile(tiling) in UngroupSharding()
1540 ? HloSharding::PartialTile(tiles) in AlignGroupsWith()
1823 return AlignShardingOnDims(HloSharding::PartialTile(tgt_tile_assignment), in CreateMatchingShardingOnDims()
1910 AlignShardingOnDims(HloSharding::PartialTile(new_tile_assignment), in GatherOperandsShardedAcrossParallelDims()
1929 ? HloSharding::PartialTile(operand_shard_tiles) in GatherOperandsShardedAcrossParallelDims()
|
| D | dot_handler.cc | 1723 ? HloSharding::PartialTile(reshaped_output_tiling) in PartitionDotGroupOnBatch()
1809 ? HloSharding::PartialTile(reshaped_tiling) in PartitionDotGroupOnBatch()
1904 ? HloSharding::PartialTile(reshaped_matching_tiling) in GetNonContractingPartitionGroupedShardingForMatchedOperand()
2118 ? HloSharding::PartialTile(new_tile) in PartitionDotGroupOnContracting()
2127 ? HloSharding::PartialTile(new_tile) in PartitionDotGroupOnContracting()
|
| D | gather_scatter_handler.cc | 367 ? HloSharding::PartialTile(output_tile_assignment) in PartitionIndexParallelDimensions()
|
| D | spmd_partitioner.cc | 1140 ? HloSharding::PartialTile(temp_target_tile) in ReshardWithAllToAll()
1292 HloSharding::PartialTile(tmp_tile_assignment); in ReshardPartialReplicateWithAllToAll()
2036 ? HloSharding::PartialTile(new_input_tile_assignment) in HandleReshape()
|