Searched refs:IsElementwise (Results 1 – 20 of 20) sorted by relevance
43 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 2)) in TEST_F()45 TF_RET_CHECK(!(hlo->IsElementwise() && hlo->operand_count() == 1)) in TEST_F()52 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 2) in TEST_F()58 TF_RET_CHECK(hlo->IsElementwise() && hlo->operand_count() == 1) in TEST_F()
47 CHECK(inst->IsElementwise()); in IsElementwiseOperationOnSimilarSlices()251 if (!instruction->IsElementwise() || instruction->operand_count() == 0 || in Run()
273 if (!instruction->IsElementwise() || instruction->operands().empty() || in IsReshapeMoveCandidate()
139 if (hlo->IsElementwise() || // in DetermineHloInstructionIsReplicated()
138 if (instr->IsElementwise() && instr->operand_count() == 1) { in ComputeArrayFor()143 } else if (instr->IsElementwise() && instr->operand_count() == 2) { in ComputeArrayFor()
1074 EXPECT_TRUE(add->IsElementwise()); in TEST_F()1094 EXPECT_TRUE(map->IsElementwise()); in TEST_F()1136 EXPECT_FALSE(fusion->IsElementwise()); in TEST_F()1177 EXPECT_FALSE(fusion->IsElementwise()); in TEST_F()
131 return !hlo->IsElementwise(); in TryRemoveConditional()
294 if (instruction->IsElementwise() && in SupportSpatialPartitioning()562 if (aggressiveness < 1 && !inst.IsElementwise() && in CanPropagateThroughAtAgressiveLevel()1024 if (instruction->IsElementwise() && may_combine_partial_sharding) { in InferShardingFromOperands()
2983 if (IsElementwise()) { in ToCategory()3448 return IsElementwise() && operand_count() == 2; in IsElementwiseBinary()3451 bool HloInstruction::IsElementwise() const { in IsElementwise() function in xla::HloInstruction3620 return IsElementwise() ? UseKind::kUse : UseKind::kReuse; in OperandElementUse()
1486 bool IsElementwise() const;
1415 if (fused->opcode() != HloOpcode::kParameter && !fused->IsElementwise()) { in IsElementwiseImpl()1435 if (user->IsElementwise() || in IsElementwiseImpl()
63 if (inst->IsElementwise()) { in ChooseIdentityValue()
678 return hlo->IsElementwise(); in IsTrivialElementwise()
3614 if (!user->IsElementwise()) { in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()5593 if (map_root->opcode() == HloOpcode::kFusion || !map_root->IsElementwise()) { in HandleMap()
218 ((instr.IsElementwise() && instr.operand_count() > 0) || in IsLoopFusible()516 instr.IsElementwise()); in IsFusibleAsMultiOutputFusionRoot()
212 } else if (hlo.IsElementwise()) { in MayPreventVectorization()
30 return hlo.IsElementwise() || // in CanBeLoopFused()
152 if (instruction->IsElementwise() || instruction->IsLoopFusion() || in GetTargetParallelTaskCount()
2893 } else if (inst->IsElementwise() && !inst->HasSideEffectNoRecurse() && in FindInputNodesIfOnlyDependOnSmallOperands()3086 inst->to_apply()->root_instruction()->IsElementwise()) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3095 inst->user_count() > 0 && inst->IsElementwise() && in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
230 if (hlo->IsElementwise() && hlo->operand_count() > 0) { in AddInstruction()1374 if (hlo->IsElementwise() && hlo->operand_count() > 0) { in DefaultAction()