Home
last modified time | relevance | path

Searched refs:IsElementwise (Results 1 – 13 of 13) 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()
Ddynamic_padder.cc46 if (inst->IsElementwise()) { in ChooseIdentityValue()
Dreshape_mover.cc272 if (!instruction->IsElementwise() || instruction->operands().empty() || in IsReshapeMoveCandidate()
Dhlo_instruction_test.cc1013 EXPECT_TRUE(add->IsElementwise()); in TEST_F()
1033 EXPECT_TRUE(map->IsElementwise()); in TEST_F()
1075 EXPECT_FALSE(fusion->IsElementwise()); in TEST_F()
1116 EXPECT_FALSE(fusion->IsElementwise()); in TEST_F()
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.cc2437 if (IsElementwise()) { in ToCategory()
2850 return IsElementwise() && operand_count() == 2; in IsElementwiseBinary()
2853 bool HloInstruction::IsElementwise() const { in IsElementwise() function in xla::HloInstruction
2966 return IsElementwise() ? UseKind::kUse : UseKind::kReuse; in OperandElementUse()
Dhlo_instruction.h1238 bool IsElementwise() const;
Dhlo_instructions.cc1182 if (fused->opcode() != HloOpcode::kParameter && !fused->IsElementwise()) { in IsElementwiseImpl()
1202 if (user->IsElementwise() || in IsElementwiseImpl()
Dalgebraic_simplifier.cc2404 if (!user->IsElementwise()) { in TryToSinkBroadcastAfterOpWithUniqueNonScalarOperand()
3714 if (map_root->opcode() == HloOpcode::kFusion || !map_root->IsElementwise()) { in HandleMap()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmulti_output_fusion.cc56 instr->IsElementwise()); in IsFusible()
161 if (!producer->IsElementwise() && !is_loop_fusion) { in DoProducerConsumerMultiOutputFusion()
Dgpu_fusible.cc163 return (instr.IsElementwise() && instr.operand_count() > 0) || in IsLoopFusible()
Dir_emitter_unnested.cc2247 if (hlo.IsElementwise() || in EmitTargetElementLoop()
3309 if (hlo->IsElementwise()) { in IsInstructionSafeForShmemTranspose()
3562 (user->IsElementwise() && in AreUsersElementwise()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion.cc33 return hlo.IsElementwise() || // in CanBeLoopFused()