Home
last modified time | relevance | path

Searched refs:has_sharding (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc66 return hlo->has_sharding() && IsSpatiallyPartitioned(hlo->sharding()); in IsSpatiallyPartitioned()
82 if (!instruction->has_sharding()) { in MaybeImproveInstructionSharding()
161 if (instruction->operand(0)->has_sharding()) { in PickRepresentativeOperand()
219 if (operand->has_sharding() && in PickRepresentativeOperand()
487 if (!inst->has_sharding()) { in InferConvolutionShardingFromOperands()
584 if (instruction->has_sharding() && instruction->sharding().IsManual()) { in InferShardingFromOperands()
590 if (!instruction->has_sharding() && in InferShardingFromOperands()
596 if (!op->has_sharding() || !op->sharding().IsManual()) continue; in InferShardingFromOperands()
611 if (op->has_sharding() && op->sharding().IsReplicated()) { in InferShardingFromOperands()
640 if (!instruction->has_sharding()) { in InferShardingFromOperands()
[all …]
Dhlo_domain_test.cc429 EXPECT_TRUE(new_tuple->has_sharding()); in TEST_F()
471 EXPECT_TRUE(root->has_sharding()); in TEST_F()
664 EXPECT_TRUE(tuple0->has_sharding()); in TEST_F()
670 EXPECT_TRUE(copy0->has_sharding()); in TEST_F()
678 EXPECT_FALSE(copy1->has_sharding()); in TEST_F()
680 EXPECT_TRUE(gte0->has_sharding()); in TEST_F()
683 EXPECT_TRUE(gte1->has_sharding()); in TEST_F()
688 EXPECT_TRUE(copy2->has_sharding()); in TEST_F()
693 EXPECT_TRUE(tuple1->has_sharding()); in TEST_F()
Dhlo_sharding_metadata.cc157 if (!instruction->has_sharding()) { in ApplyDomainSingleSharding()
263 if (!user->has_sharding()) { in ApplyShardingFromUsers()
324 if (instruction->has_sharding()) { in ApplyDomainShardingPass()
356 if (!instruction->has_sharding()) { in ApplyDomainSharding()
383 if (instruction->has_sharding()) { in ExtractOriginalCommonSharding()
Dhlo_matchers.cc190 if (!instruction->has_sharding()) { in MatchAndExplain()
196 if (instruction->has_sharding()) { in MatchAndExplain()
Dbatchnorm_expander.cc268 if (batch_norm->has_sharding()) { in HandleBatchNormTraining()
360 if (batch_norm->has_sharding()) { in HandleBatchNormInference()
532 if (batch_norm->has_sharding()) { in HandleBatchNormGrad()
Dhlo_replication_analysis.cc299 if (inst->has_sharding()) { in ComputeHloReplicationOnComputation()
335 if (cross_partition_spmd_ && param->has_sharding()) { in ComputeHloReplication()
Dhlo_instruction.h1383 CHECK(has_sharding()); in sharding()
1415 bool has_sharding() const { return sharding_ != nullptr; } in has_sharding() function
1419 if (!has_sharding()) { in has_compatible_sharding()
1420 return !other->has_sharding(); in has_compatible_sharding()
1422 return other->has_sharding() ? sharding() == other->sharding() : false; in has_compatible_sharding()
Dconditional_simplifier.cc255 if (conditional->has_sharding()) { in TryRemoveUnusedConditionalOperands()
591 if (conditional_op->has_sharding()) { in Run()
Dall_reduce_combiner.cc91 if (to_combine.front()->has_sharding()) { in CombineAllReduces()
Dhlo_sharding_util.cc217 if (!instruction->has_sharding()) { in AssignComputationDevice()
229 if (instruction->has_sharding()) { in GetMostOccurringDevice()
246 if (instruction->has_sharding()) { in GetDominantDevice()
490 if (instruction->has_sharding() && in ContainsTileSharding()
Dhlo_instruction.cc810 if (proto.has_sharding()) { in CreateFromProto()
1398 if (operand->has_sharding()) { in CreateBroadcastSequence()
1424 if (operand->has_sharding()) { in CreateBroadcastSequence()
1432 if (operand->has_sharding()) { in CreateBroadcastSequence()
2903 if (has_sharding()) { in ExtraAttributesToString()
2962 if (has_sharding()) { in ToProto()
Dhlo_instruction_test.cc788 EXPECT_FALSE(tuple_clone->has_sharding()); in TEST_F()
807 EXPECT_FALSE(tuple_clone->has_sharding()); in TEST_F()
Dhlo_computation.cc939 if (!new_instruction->has_sharding()) { in ReplaceInstruction()
Dsharding_propagation_test.cc41 if (!instruction->has_sharding()) { in ClearMetadata()
1678 ASSERT_TRUE(instr->has_sharding()); in TEST_P()
1798 ASSERT_TRUE(instr->has_sharding()); in TEST_P()
Dhlo_graph_dumper.cc906 if (!instr->has_sharding()) { in GetInstructionColor()
Dlayout_assignment.cc911 if (instruction.has_sharding()) { in SetupCopiedInstruction()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dutils.cc63 if (instr.has_sharding()) { in GetShardedShape()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc105 return !hlo->has_sharding() || in ReportBeforePartition()
1806 if (!hlo->operand(0)->has_sharding()) { in HandleCustomCall()
3661 CHECK(param->has_sharding()) << "Missing sharding in entry parameter " << i; in Run()
3666 CHECK(entry_root->has_sharding()) << "Missing sharding in entry root."; in Run()
3737 TF_RET_CHECK(hlo->has_sharding()) in PreprocessSharding()
3755 if (!hlo->has_sharding()) { in PreprocessSharding()
3782 TF_RET_CHECK(entry->root_instruction()->has_sharding()); in PreprocessSharding()
3789 TF_RET_CHECK(param->has_sharding()); in PreprocessSharding()
Dfft_handler.cc364 if (!hlo->has_sharding() || in HandleFft()
Dspmd_partitioner.h295 CHECK(hlo->has_sharding()) in PartitionedHlo()
Dspmd_partitioner_util.cc1224 if (!data->has_sharding()) { in GetKValueInTopKWhenPartitionSortDim()