Home
last modified time | relevance | path

Searched refs:fused_expression_root (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc66 instr.fused_expression_root()->operands()) { in IsReduceInputFusion()
76 IsReductionToVector(*instr.fused_expression_root())) { in IsReduceInputFusion()
96 auto fused_expression_root = instr->fused_expression_root(); in ShapesCompatibleForMultiOutputFusion() local
100 for (const auto* inst : fused_expression_root->operands()) { in ShapesCompatibleForMultiOutputFusion()
105 return fused_expression_root->operands()[0]; in ShapesCompatibleForMultiOutputFusion()
107 return fused_expression_root; in ShapesCompatibleForMultiOutputFusion()
145 instr.fused_expression_root()->opcode() == HloOpcode::kScatter)) { in IsInputFusibleScatter()
Dmulti_output_fusion_test.cc72 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
160 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
193 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
255 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
299 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
355 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
403 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
430 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
471 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
538 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
Dinstruction_fusion_test.cc171 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
244 EXPECT_THAT(root->fused_expression_root(), op::Add(op::Copy(), op::Copy())); in TEST_F()
265 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
287 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
329 root->fused_expression_root(), in TEST_F()
356 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
463 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
536 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
606 EXPECT_THAT(root->operand(0)->fused_expression_root(), in TEST_F()
657 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
Dgpu_fusible_test.cc85 ASSERT_EQ(reduce_fusion->fused_expression_root()->opcode(), in TEST_F()
89 ASSERT_EQ(loop_fusion->fused_expression_root()->opcode(), HloOpcode::kSelect); in TEST_F()
110 ASSERT_EQ(reduce->fused_expression_root()->opcode(), HloOpcode::kReduce); in TEST_F()
145 ASSERT_EQ(reduce_fusion->fused_expression_root()->opcode(), in TEST_F()
149 ASSERT_EQ(loop_fusion->fused_expression_root()->opcode(), HloOpcode::kCopy); in TEST_F()
177 ASSERT_EQ(loop_fusion->fused_expression_root()->opcode(), HloOpcode::kAdd); in TEST_F()
Dir_emission_utils.cc94 (hlo.fused_expression_root()->opcode() == HloOpcode::kMultiply || in ImplementedAsGemm()
95 hlo.fused_expression_root()->opcode() == HloOpcode::kAdd)) { in ImplementedAsGemm()
97 const HloInstruction* dot = hlo.fused_expression_root()->operand(0); in ImplementedAsGemm()
99 dot = hlo.fused_expression_root()->operand(1); in ImplementedAsGemm()
Dinstruction_fusion.cc124 producer->fused_expression_root()->opcode() == HloOpcode::kDot)) { in ShouldFuseInexpensiveChecks()
131 Match(consumer->fused_expression_root(), in ShouldFuseInexpensiveChecks()
287 producer->fused_expression_root()->opcode() == HloOpcode::kDot)) { in ChooseKind()
Dgemm_thunk.cc277 CHECK(hlo_instruction.fused_expression_root()->opcode() == HloOpcode::kAdd || in GetDimensionNumbers()
278 hlo_instruction.fused_expression_root()->opcode() == in GetDimensionNumbers()
282 hlo_instruction.fused_expression_root()->operand(0); in GetDimensionNumbers()
284 dot = hlo_instruction.fused_expression_root()->operand(1); in GetDimensionNumbers()
Dfusion_merger.cc106 for (auto& operand : fusion->fused_expression_root()->operands()) { in GetCurrentBytesTransferred()
111 ShapeUtil::ByteSizeOf(fusion->fused_expression_root()->shape()); in GetCurrentBytesTransferred()
Dmulti_output_fusion.cc104 if (instr1->fused_expression_root()->opcode() == in LegalToFuse()
107 instr2->fused_expression_root()->opcode() == in LegalToFuse()
Dir_emitter_unnested.cc576 HloInstruction* root = fusion->fused_expression_root(); in HandleFusion()
641 << fusion->fused_expression_root()->opcode(); in HandleFusion()
1793 const HloInstruction* output_fused_op = inst->fused_expression_root(); in BuildGemmThunk()
1910 HloInstruction* inst = fused ? hlo->fused_expression_root() : hlo; in BuildInitializerThunk()
2499 TF_CHECK_OK(hlo->fused_expression_root()->Accept(&fused_emitter)); in EmitTileElementForFusion()
2702 ? unnested_hlo->fused_expression_root() in EmitPrologueForReduction()
2858 ? unnested_hlo->fused_expression_root() in EmitTileElementForReduction()
2893 TF_CHECK_OK(unnested_hlo->fused_expression_root()->Accept(&fused_emitter)); in EmitTileElementForReduction()
3613 const HloInstruction* fused_root = unnested_hlo->fused_expression_root(); in IsUnrollingColumnReductionBeneficial()
3733 ? unnested_hlo->fused_expression_root() in EmitReductionToVector()
/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion_test.cc64 EXPECT_THAT(fusion->fused_expression_root(), in TEST_F()
88 EXPECT_THAT(fusion->fused_expression_root(), op::Add(op::Abs(), op::Abs())) in TEST_F()
109 EXPECT_THAT(fusion->fused_expression_root(), op::Tuple(op::Tanh(), op::Abs())) in TEST_F()
243 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
343 EXPECT_THAT(root->fused_expression_root(), in TEST_F()
Dreduce_precision_insertion_test.cc417 HloInstruction* y_fused = z->fused_expression_root(); in TEST_F()
431 EXPECT_EQ(z->fused_expression_root(), y_fused); in TEST_F()
453 HloInstruction* y_fused = z->fused_expression_root(); in TEST_F()
477 EXPECT_THAT(z->fused_expression_root(), y_fused); in TEST_F()
500 HloInstruction* y_fused = z->fused_expression_root(); in TEST_F()
524 EXPECT_THAT(z->fused_expression_root(), op::ReducePrecision(y_fused)); in TEST_F()
Dtuple_points_to_analysis.cc167 TF_RETURN_IF_ERROR(instruction->fused_expression_root()->Accept(this)); in Analyze()
685 fused_param_uses[0].first == fusion->fused_expression_root() && in HasUniqueFusedUseOfOperandAt()
722 if (user->fused_expression_root()->opcode() == in CanShareOperandBufferWithUser()
737 user->fused_expression_root()->opcode() == HloOpcode::kAdd) { in CanShareOperandBufferWithUser()
741 auto* add = user->fused_expression_root(); in CanShareOperandBufferWithUser()
Ddefuser.cc82 defused_instructions.at(fusion_instruction->fused_expression_root()))); in Defuse()
Dhlo_dataflow_analysis.cc976 Cast<HloDynamicUpdateSliceInstruction>(fusion->fused_expression_root()); in CanDoInPlaceDynamicUpdateSlice()
1050 if (user->fused_expression_root()->opcode() == in CanShareOperandBufferWithUser()
1065 user->fused_expression_root()->opcode() == HloOpcode::kAdd) { in CanShareOperandBufferWithUser()
1069 auto* add = user->fused_expression_root(); in CanShareOperandBufferWithUser()
1087 return use.instruction == user->fused_expression_root() && in CanShareOperandBufferWithUser()
Dlogical_buffer_analysis.cc73 TF_RETURN_IF_ERROR(instruction->fused_expression_root()->Accept(this)); in Analyze()
Dreduce_precision_insertion.cc180 HloInstruction* fusion_root = instruction->fused_expression_root(); in insert_on_outputs()
Ddefuser_test.cc205 HloInstruction* fused_negate = outer_fusion->fused_expression_root(); in TEST_F()
Dhlo_instruction_test.cc703 metadata, fusion->fused_expression_root()->metadata())); in TEST_F()
705 metadata, fusion->fused_expression_root()->operand(0)->metadata())); in TEST_F()
1161 const HloInstruction* root = fusion->fused_expression_root(); in TEST_F()
1162 const HloInstruction* root2 = fusion2->fused_expression_root(); in TEST_F()
Dhlo_instructions.cc1256 CHECK(unfused_instructions.front() == cloned_fusion->fused_expression_root()); in MergeFusionInstruction()
1336 HloInstruction* HloFusionInstruction::fused_expression_root() const { in fused_expression_root() function in xla::HloFusionInstruction
1392 clone = fused_expression_root(); in CloneAndFuseInternal()
1475 HloInstruction* fused_root = fused_expression_root(); in CloneAndFuseInternal()
Dhlo_instructions.h761 HloInstruction* fused_expression_root() const;
786 return fused_expression_root()->opcode() == HloOpcode::kTuple; in IsMultiOutputFusion()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc79 fusion_instruction->fused_expression_root()->opcode()); in TEST_F()
126 fusion_instruction->fused_expression_root()->opcode()); in TEST_F()
202 fusion_instruction1->fused_expression_root()->opcode()); in TEST_F()
211 fusion_instruction2->fused_expression_root()->opcode()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Ddynamic_update_slice_util.h46 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace()
Ddynamic_update_slice_util.cc146 auto* dynamic_update_slice = fusion->fused_expression_root(); in EmitFusedDynamicUpdateSliceInPlaceImpl()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dinteractive_graphviz.cc321 << instr->fused_expression_root()->name() << std::endl; in DoInfoCommand()

12