Home
last modified time | relevance | path

Searched refs:IsElementwise (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default_test.cc43 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()
Dslice_sinker.cc47 CHECK(inst->IsElementwise()); in IsElementwiseOperationOnSimilarSlices()
251 if (!instruction->IsElementwise() || instruction->operand_count() == 0 || in Run()
Dreshape_mover.cc273 if (!instruction->IsElementwise() || instruction->operands().empty() || in IsReshapeMoveCandidate()
Dhlo_replication_analysis.cc139 if (hlo->IsElementwise() || // in DetermineHloInstructionIsReplicated()
Dindexed_array_analysis.cc138 if (instr->IsElementwise() && instr->operand_count() == 1) { in ComputeArrayFor()
143 } else if (instr->IsElementwise() && instr->operand_count() == 2) { in ComputeArrayFor()
Dhlo_instruction_test.cc1074 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()
Dconditional_simplifier.cc131 return !hlo->IsElementwise(); in TryRemoveConditional()
Dsharding_propagation.cc294 if (instruction->IsElementwise() && in SupportSpatialPartitioning()
562 if (aggressiveness < 1 && !inst.IsElementwise() && in CanPropagateThroughAtAgressiveLevel()
1024 if (instruction->IsElementwise() && may_combine_partial_sharding) { in InferShardingFromOperands()
Dhlo_instruction.cc2983 if (IsElementwise()) { in ToCategory()
3448 return IsElementwise() && operand_count() == 2; in IsElementwiseBinary()
3451 bool HloInstruction::IsElementwise() const { in IsElementwise() function in xla::HloInstruction
3620 return IsElementwise() ? UseKind::kUse : UseKind::kReuse; in OperandElementUse()
Dhlo_instruction.h1486 bool IsElementwise() const;
Dhlo_instructions.cc1415 if (fused->opcode() != HloOpcode::kParameter && !fused->IsElementwise()) { in IsElementwiseImpl()
1435 if (user->IsElementwise() || in IsElementwiseImpl()
Ddynamic_padder.cc63 if (inst->IsElementwise()) { in ChooseIdentityValue()
Dspace_to_batch_converter.cc678 return hlo->IsElementwise(); in IsTrivialElementwise()
Dalgebraic_simplifier.cc3614 if (!user->IsElementwise()) { in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
5593 if (map_root->opcode() == HloOpcode::kFusion || !map_root->IsElementwise()) { in HandleMap()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc218 ((instr.IsElementwise() && instr.operand_count() > 0) || in IsLoopFusible()
516 instr.IsElementwise()); in IsFusibleAsMultiOutputFusionRoot()
Dir_emitter_unnested.cc212 } else if (hlo.IsElementwise()) { in MayPreventVectorization()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc30 return hlo.IsElementwise() || // in CanBeLoopFused()
Dparallel_task_assignment.cc152 if (instruction->IsElementwise() || instruction->IsLoopFusion() || in GetTargetParallelTaskCount()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2893 } 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()
Dspmd_partitioner.cc230 if (hlo->IsElementwise() && hlo->operand_count() > 0) { in AddInstruction()
1374 if (hlo->IsElementwise() && hlo->operand_count() > 0) { in DefaultAction()