Home
last modified time | relevance | path

Searched refs:GpuInstructionFusion (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc47 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
67 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
86 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
105 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
123 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
140 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
165 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
193 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
223 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
238 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
[all …]
Dinstruction_fusion.cc43 /*static*/ bool GpuInstructionFusion::IsExpensive( in IsExpensive()
74 /*static*/ bool GpuInstructionFusion::FusionWouldBeTooLarge( in FusionWouldBeTooLarge()
117 bool GpuInstructionFusion::ShouldFuseInexpensiveChecks(HloInstruction* consumer, in ShouldFuseInexpensiveChecks()
257 bool GpuInstructionFusion::ShouldFuse(HloInstruction* consumer, in ShouldFuse()
274 bool GpuInstructionFusion::ShouldFuseIntoMultiOutput(HloInstruction* consumer, in ShouldFuseIntoMultiOutput()
279 HloInstruction::FusionKind GpuInstructionFusion::ChooseKind( in ChooseKind()
Dinstruction_fusion.h25 class GpuInstructionFusion : public InstructionFusion {
27 explicit GpuInstructionFusion(bool may_duplicate) in GpuInstructionFusion() function
28 : InstructionFusion(GpuInstructionFusion::IsExpensive, may_duplicate) {} in GpuInstructionFusion()
Dmulti_output_fusion_test.cc574 ASSERT_GT(kNumParams, GpuInstructionFusion::kMaxOperandsAndOutputsPerFusion); in TEST_F()
619 GpuInstructionFusion::kMaxOperandsAndOutputsPerFusion) in TEST_F()
Dfusion_merger.cc229 GpuInstructionFusion::IsExpensive(*instruction); in HandleFusion()
Dmulti_output_fusion.cc113 return !GpuInstructionFusion::FusionWouldBeTooLarge(instr1, instr2); in LegalToFuse()
Dnvptx_compiler.cc365 fusion.AddPass<GpuInstructionFusion>(/*may_duplicate=*/false); in OptimizeHloModule()
366 fusion.AddPass<GpuInstructionFusion>(/*may_duplicate=*/true); in OptimizeHloModule()