/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_fusible.cc | 66 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()
|
D | multi_output_fusion_test.cc | 72 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()
|
D | instruction_fusion_test.cc | 171 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()
|
D | gpu_fusible_test.cc | 85 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()
|
D | ir_emission_utils.cc | 94 (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()
|
D | instruction_fusion.cc | 124 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()
|
D | gemm_thunk.cc | 277 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()
|
D | fusion_merger.cc | 106 for (auto& operand : fusion->fused_expression_root()->operands()) { in GetCurrentBytesTransferred() 111 ShapeUtil::ByteSizeOf(fusion->fused_expression_root()->shape()); in GetCurrentBytesTransferred()
|
D | multi_output_fusion.cc | 104 if (instr1->fused_expression_root()->opcode() == in LegalToFuse() 107 instr2->fused_expression_root()->opcode() == in LegalToFuse()
|
D | ir_emitter_unnested.cc | 576 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/ |
D | instruction_fusion_test.cc | 64 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()
|
D | reduce_precision_insertion_test.cc | 417 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()
|
D | tuple_points_to_analysis.cc | 167 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()
|
D | defuser.cc | 82 defused_instructions.at(fusion_instruction->fused_expression_root()))); in Defuse()
|
D | hlo_dataflow_analysis.cc | 976 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()
|
D | logical_buffer_analysis.cc | 73 TF_RETURN_IF_ERROR(instruction->fused_expression_root()->Accept(this)); in Analyze()
|
D | reduce_precision_insertion.cc | 180 HloInstruction* fusion_root = instruction->fused_expression_root(); in insert_on_outputs()
|
D | defuser_test.cc | 205 HloInstruction* fused_negate = outer_fusion->fused_expression_root(); in TEST_F()
|
D | hlo_instruction_test.cc | 703 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()
|
D | hlo_instructions.cc | 1256 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()
|
D | hlo_instructions.h | 761 HloInstruction* fused_expression_root() const; 786 return fused_expression_root()->opcode() == HloOpcode::kTuple; in IsMultiOutputFusion()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_fusion_test.cc | 79 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/ |
D | dynamic_update_slice_util.h | 46 HloInstruction* fused_root = fusion->fused_expression_root(); in CanEmitFusedDynamicUpdateSliceInPlace()
|
D | dynamic_update_slice_util.cc | 146 auto* dynamic_update_slice = fusion->fused_expression_root(); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | interactive_graphviz.cc | 321 << instr->fused_expression_root()->name() << std::endl; in DoInfoCommand()
|