Home
last modified time | relevance | path

Searched refs:IsTileMaximal (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc57 if (!rhs.IsTileMaximal()) { in IsShardingMoreSpecific()
62 return !lhs.IsTileMaximal(); in IsShardingMoreSpecific()
270 if (sharding.IsTileMaximal()) { in TransposeSharding()
300 if (sharding.IsTileMaximal()) { in ReshapeSharding()
414 if (sharding.IsTileMaximal() || dimensions.empty()) { in ReverseSharding()
435 CHECK(!sharding.IsTuple() && !sharding.IsTileMaximal()); in ReshapeToTileDimension()
491 !instruction->sharding().IsTileMaximal()) { in ContainsTileSharding()
501 if (index_sharding.IsTileMaximal()) { in GatherOutputSharding()
540 if (output_sharding.IsTileMaximal()) { in GatherIndexSharding()
589 if (hlo.sharding().IsTileMaximal()) { in GatherEffectiveOutputSharding()
[all …]
Dhlo_sharding.cc449 if (IsTileMaximal() || IsManual()) { in ValidateNonTuple()
555 } else if (IsTileMaximal()) { in ToProto()
568 if (IsTileMaximal() || IsManual()) { in TileShape()
580 if (IsTileMaximal() || IsManual()) { in TileShape()
599 if (IsTileMaximal()) { in NumTiles()
611 if (IsTileMaximal()) { in NumTiles()
Dhlo_sharding.h135 bool IsTileMaximal() const { in IsTileMaximal() function
140 return s.IsTileMaximal(); in IsTileMaximal()
Dsharding_propagation.cc62 return !sharding.IsTileMaximal() || sharding.IsReplicated(); in IsSpatiallyPartitioned()
93 !instruction->sharding().IsTileMaximal() && in MaybeImproveInstructionSharding()
343 if (operand_sharding.IsTileMaximal()) { in InferDotShardingFromOperands()
411 if (operand_sharding.IsTileMaximal()) { in InferGatherParallelShardingFromOperands()
436 if (replicate_non_parallel_dims.IsTileMaximal()) { in InferGatherParallelShardingFromOperands()
491 if (sharding.IsTileMaximal()) { in InferConvolutionShardingFromOperands()
520 CHECK(!lhs->sharding().IsTileMaximal()); in InferConvolutionShardingFromOperands()
892 if (!operand->sharding().IsTileMaximal() && in InferShardingFromOperands()
1318 if (user_sharding.IsTileMaximal()) { in GetShardingFromUser()
1380 if (!from_indices.IsTileMaximal()) { in GetShardingFromUser()
Dhlo_sharding_test.cc59 EXPECT_TRUE(sharding.IsTileMaximal()); in TEST_F()
74 EXPECT_TRUE(sharding.IsTileMaximal()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc418 if (!target.IsTileMaximal() && sharding().ReplicateOnLastTileDim()) { in ReshardNoCache()
429 if (!sharding().IsTileMaximal() && target.ReplicateOnLastTileDim()) { in ReshardNoCache()
447 if (target.IsTileMaximal()) { in ReshardNoCache()
488 CHECK(!sharding.IsTileMaximal()); in PadWithValue()
565 CHECK(!target.IsTileMaximal()); in ReshardAsWindowedInput()
854 if (sharding.IsTileMaximal()) { in Replicate()
867 CHECK(!sharding().IsTileMaximal()); in ReplicatePartial()
1239 if (tile_sharding.ReplicateOnLastTileDim() || tile_sharding.IsTileMaximal()) { in ReshardPartialReplicateWithAllToAll()
1378 if (!hlo->sharding().IsTileMaximal()) { in DefaultAction()
1487 if (sharding.IsTileMaximal()) { in HandleConcatenate()
[all …]
Dgather_scatter_handler.cc37 if (operand.sharding().IsTileMaximal()) { in GatherScatterOperandPartitionedOnlyOnTrivialSliceDims()
140 if (operand.sharding().IsTileMaximal()) { in PartitionIndexOnlyPartition()
141 if (!indices.sharding().IsTileMaximal() && in PartitionIndexOnlyPartition()
486 if (!operand.sharding().IsTileMaximal()) { in PartitionGather()
531 if (operand.sharding().IsTileMaximal()) { in HandleScatter()
532 if (!indices.sharding().IsTileMaximal() && in HandleScatter()
Dconvolution_handler.cc236 TF_RET_CHECK(!lhs.sharding().IsTileMaximal() && in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
237 !rhs.sharding().IsTileMaximal()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
523 TF_RET_CHECK(!lhs.sharding().IsTileMaximal() && in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
524 !rhs.sharding().IsTileMaximal()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
749 TF_RET_CHECK(!output_sharding.IsTileMaximal()); in PartitionConvolutionTiledOutput()
856 if (!lhs.sharding().IsTileMaximal() && !rhs.sharding().IsTileMaximal()) { in PartitionConvolutionBaseCase()
880 if (!output_sharding.IsTileMaximal()) { in PartitionConvolutionBaseCase()
Dspmd_partitioner_util.cc137 if (sharding.IsTileMaximal()) { in EvenlyPartitions()
182 if (sharding.IsTileMaximal()) { in MakeNonPaddedShapeForGivenPartition()
241 CHECK(!sharding.IsTileMaximal()); in MakeTiledPartitionOrdinals()
278 if (sharding.IsTileMaximal()) { in GetPaddedShapeForUnevenPartitioning()
630 if (sharding.IsTileMaximal()) { in UniqueTiledDim()
1073 if (original.sharding().IsTileMaximal()) { in HaloExchangeToPadOnLeft()
1229 if (sharding.IsTileMaximal()) { in GetKValueInTopKWhenPartitionSortDim()
1277 if (sharding.IsTileMaximal()) { in ShardCountAtDim()
1286 if (source.IsTileMaximal() || target.IsTileMaximal() || in GetReshardAllToAllSourceTargetDims()
1371 return !source.IsTileMaximal() && !target.IsTileMaximal() && in CanReshardWithCollectivePermute()
[all …]
Ddot_handler.cc1753 if (operand.sharding().IsTileMaximal()) { in PartitionDotGroupOnBatch()
2457 if (sharding.IsTileMaximal()) { in PartitionDot()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc143 if (sharding && !sharding->IsTileMaximal() && !sharding->IsManual()) { in RewriteLayoutWithShardedShape()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc181 if (sharding.IsTileMaximal()) { in GetPerDeviceShape()