Searched refs:has_sharding (Results 1 – 8 of 8) sorted by relevance
429 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()
157 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()
186 if (!instruction->has_sharding()) { in MatchAndExplain()192 if (instruction->has_sharding()) { in MatchAndExplain()
1140 CHECK(has_sharding()); in sharding()1172 bool has_sharding() const { return sharding_ != nullptr; } in has_sharding() function1176 if (!has_sharding()) { in has_compatible_sharding()1177 return !other->has_sharding(); in has_compatible_sharding()1179 return other->has_sharding() ? sharding() == other->sharding() : false; in has_compatible_sharding()
301 if (batch_norm->has_sharding()) { in HandleBatchNormTraining()410 if (batch_norm->has_sharding()) { in HandleBatchNormInference()583 if (batch_norm->has_sharding()) { in HandleBatchNormGrad()
628 if (proto.has_sharding()) { in CreateFromProto()1129 if (operand->has_sharding()) { in CreateBroadcastSequence()1154 if (operand->has_sharding()) { in CreateBroadcastSequence()1161 if (operand->has_sharding()) { in CreateBroadcastSequence()2375 if (has_sharding()) { in ExtraAttributesToString()2424 if (has_sharding()) { in ToProto()
891 if (!instr->has_sharding()) { in GetInstructionColor()
858 if (instruction.has_sharding()) { in SetupCopiedInstruction()