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.cc48 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
68 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
87 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
106 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
126 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
143 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
168 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
196 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
226 EXPECT_FALSE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
241 EXPECT_TRUE(GpuInstructionFusion(/*may_duplicate=*/true) in TEST_F()
[all …]
Dinstruction_fusion.cc39 /*static*/ bool GpuInstructionFusion::IsExpensive( in IsExpensive()
57 bool GpuInstructionFusion::ShouldFuseInexpensiveChecks(HloInstruction* consumer, in ShouldFuseInexpensiveChecks()
84 bool GpuInstructionFusion::ShouldFuse(HloInstruction* consumer, in ShouldFuse()
123 bool GpuInstructionFusion::ShouldFuseIntoMultiOutput(HloInstruction* consumer, in ShouldFuseIntoMultiOutput()
128 HloInstruction::FusionKind GpuInstructionFusion::ChooseKind( in ChooseKind()
133 HloInstruction* GpuInstructionFusion::FuseInstruction( in FuseInstruction()
Dinstruction_fusion.h27 class GpuInstructionFusion : public InstructionFusion {
29 explicit GpuInstructionFusion(bool may_duplicate) in GpuInstructionFusion() function
30 : InstructionFusion(GpuInstructionFusion::IsExpensive, may_duplicate) {} in GpuInstructionFusion()
Dhorizontal_loop_fusion_test.cc190 fusion.AddPass<xla::gpu::GpuInstructionFusion>(/*may_duplicate=*/false); in TEST_F()
191 fusion.AddPass<xla::gpu::GpuInstructionFusion>(/*may_duplicate=*/true); in TEST_F()
Dfusion_merger.cc257 GpuInstructionFusion::IsExpensive(*instruction); in HandleFusion()
Dgpu_compiler.cc320 fusion.AddPass<GpuInstructionFusion>(/*may_duplicate=*/false); in OptimizeHloModule()
321 fusion.AddPass<GpuInstructionFusion>(/*may_duplicate=*/true); in OptimizeHloModule()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_fusion_test.cc82 EXPECT_TRUE(GpuInstructionFusion(false).Run(module.get()).ValueOrDie()); in TEST_F()