Home
last modified time | relevance | path

Searched refs:GpuHorizontalInputFusion (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.h40 class GpuHorizontalInputFusion : public HloModulePass {
42 GpuHorizontalInputFusion() {} in GpuHorizontalInputFusion() function
Dhorizontal_input_fusion.cc150 StatusOr<bool> GpuHorizontalInputFusion::RunOnComputation( in RunOnComputation()
156 StatusOr<bool> GpuHorizontalInputFusion::Run(HloModule* module) { in Run()
Dhorizontal_input_fusion_test.cc67 EXPECT_TRUE(GpuHorizontalInputFusion().Run(module.get()).ValueOrDie()); in TEST_F()
211 EXPECT_TRUE(GpuHorizontalInputFusion().Run(module.get()).ValueOrDie()); in TEST_F()
Dgpu_compiler.cc331 horizontal_fusion.AddPass<GpuHorizontalInputFusion>(); in OptimizeHloModule()