add find algorithm for convolution without extra padding (#1710)

d38aba91 · Fenglei · Robert Kimball · 2aa7899c · d38aba91 · d38aba91
Commit d38aba91 authored Oct 01, 2018 by Fenglei Committed by Robert Kimball Oct 01, 2018
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

cudnn_emitter.cpp src/ngraph/runtime/gpu/cudnn_emitter.cpp +0 -0

cudnn_emitter.hpp src/ngraph/runtime/gpu/cudnn_emitter.hpp +6 -3

No files found.
--- a/src/ngraph/runtime/gpu/cudnn_emitter.cpp
+++ b/src/ngraph/runtime/gpu/cudnn_emitter.cpp
--- a/src/ngraph/runtime/gpu/cudnn_emitter.hpp
+++ b/src/ngraph/runtime/gpu/cudnn_emitter.hpp
@@ -76,7 +76,8 @@ namespace ngraph
                                         const Shape& output_tensor_shape,
                                         const Strides& window_movement_strides,
                                         const Strides& window_dilation_strides,
-                                         const Shape& padding_below);
+                                         const Shape& padding_below,
+                                         const bool find_algo = false);
                size_t build_convolution_backward_data(const std::string& dtype,
                                                       const Shape& input_filter_shape,
@@ -84,7 +85,8 @@ namespace ngraph
                                                       const Shape& output_tensor_shape,
                                                       const Strides& window_movement_strides,
                                                       const Strides& window_dilation_strides,
-                                                       const Shape& padding_below);
+                                                       const Shape& padding_below,
+                                                       const bool find_algo = false);
                size_t build_convolution_backward_filter(const std::string& dtype,
                                                         const Shape& input_tensor_shape_0,
@@ -92,7 +94,8 @@ namespace ngraph
                                                         const Shape& output_filter_shape,
                                                         const Strides& window_movement_strides,
                                                         const Strides& window_dilation_strides,
-                                                         const Shape& padding_below);
+                                                         const Shape& padding_below,
+                                                         const bool find_algo = false);
                size_t build_reduce_forward(const cudnnReduceTensorOp_t& reduce_op,
                                            const std::string& dtype,