Searched refs:IsMultiOutputFusion (Results 1 – 10 of 10) sorted by relevance
71 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()159 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()192 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()254 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()298 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()354 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()402 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()429 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()470 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()537 ASSERT_TRUE(fusion->IsMultiOutputFusion()); in TEST_F()
64 if (instr.IsMultiOutputFusion()) { in IsReduceInputFusion()97 if (instr->IsMultiOutputFusion()) { in ShapesCompatibleForMultiOutputFusion()
105 if (fusion->IsMultiOutputFusion()) { in GetCurrentBytesTransferred()199 if (fusion->IsMultiOutputFusion()) { in HandleFusion()
121 if (hlo.IsMultiOutputFusion()) { in EmitTargetElementLoop()
271 const Shape& element_shape = hlo->IsMultiOutputFusion() in ComputeMaxUnrollFactor()1918 CHECK(hlo->IsMultiOutputFusion()) in BuildInitializerThunk()2180 const Shape& element_shape = hlo.IsMultiOutputFusion() in EmitTargetElementLoopInThunk()2190 if (!hlo.IsMultiOutputFusion()) { in EmitTargetElementLoopInThunk()2506 if (hlo->IsMultiOutputFusion()) { in EmitTileElementForFusion()3096 if (!reduction_info && unnested_hlo->IsMultiOutputFusion()) { in EmitKernel()3372 const Shape& output_shape = hlo->IsMultiOutputFusion() in CheckAndEmitHloWithTile021()
122 if (fused->IsMultiOutputFusion()) { in Fuse()235 if (!instr->IsMultiOutputFusion()) { in LegalToFuse()
1477 const bool IsMultiOutputFusion() const;
785 const bool IsMultiOutputFusion() const { in IsMultiOutputFusion() function
3350 const bool HloInstruction::IsMultiOutputFusion() const { in IsMultiOutputFusion() function in xla::HloInstruction3352 return fusion != nullptr && fusion->IsMultiOutputFusion(); in IsMultiOutputFusion()
2957 if (target_op->IsMultiOutputFusion()) { in EmitTargetElementLoop()