Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions_test.cc46 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
71 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
94 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
115 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
137 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
159 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
180 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
205 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
231 EXPECT_TRUE(CudnnPadForConvolutions(true).Run(module.get()).ValueOrDie()); in TEST_F()
Dcudnn_pad_for_convolutions.h32 class CudnnPadForConvolutions : public HloModulePass {
34 explicit CudnnPadForConvolutions(bool is_volta_or_later) in CudnnPadForConvolutions() function
Dnvptx_compiler.cc120 pipeline.AddPass<CudnnPadForConvolutions>(IsVoltaOrLater(*stream_exec)); in OptimizeHloConvolutionCanonicalization()
Dcudnn_pad_for_convolutions.cc387 StatusOr<bool> CudnnPadForConvolutions::Run(HloModule* module) { in Run()