未验证 提交 689e27af 编写于 作者: L lzydev 提交者: GitHub

fix bug when place 'use_cudnn' in extra (#54766)

上级 610a47dd
...@@ -1979,7 +1979,7 @@ ...@@ -1979,7 +1979,7 @@
pool2d_double_grad : GetPoolDoubleGradExpectedKernelType pool2d_double_grad : GetPoolDoubleGradExpectedKernelType
extra : extra :
attrs : [bool use_mkldnn = false, bool use_quantizer = false, attrs : [bool use_mkldnn = false, bool use_quantizer = false,
str mkldnn_data_type = "float32", bool is_test = false, bool use_cudnn = false] str mkldnn_data_type = "float32", bool is_test = false]
- op : pool3d - op : pool3d
backward : pool3d_grad backward : pool3d_grad
...@@ -1993,7 +1993,7 @@ ...@@ -1993,7 +1993,7 @@
pool3d : GetPoolExpectedKernelType pool3d : GetPoolExpectedKernelType
pool3d_grad : GetPoolExpectedKernelType pool3d_grad : GetPoolExpectedKernelType
extra : extra :
attrs : [bool use_mkldnn = false, bool use_cudnn = false] attrs : [bool use_mkldnn = false]
- op : pow - op : pow
backward : pow_grad, pow_double_grad, pow_triple_grad backward : pow_grad, pow_double_grad, pow_triple_grad
......
...@@ -412,7 +412,7 @@ ...@@ -412,7 +412,7 @@
param : [peer, dtype, out_shape] param : [peer, dtype, out_shape]
- op : pool2d - op : pool2d
args : (Tensor x, IntArray kernel_size, int[] strides = {1,1}, int[] paddings = {0,0}, bool ceil_mode = false, bool exclusive = true, str data_format = "NCHW", str pooling_type = "", bool global_pooling = false, bool adaptive = false, str padding_algorithm = "EXPLICIT") args : (Tensor x, IntArray kernel_size, int[] strides = {1,1}, int[] paddings = {0,0}, bool ceil_mode = false, bool exclusive = true, str data_format = "NCHW", str pooling_type = "", bool global_pooling = false, bool adaptive = false, str padding_algorithm = "EXPLICIT", bool use_cudnn = false)
output : Tensor(out) output : Tensor(out)
infer_meta : infer_meta :
func : Pool2DInferMeta func : Pool2DInferMeta
...@@ -423,7 +423,7 @@ ...@@ -423,7 +423,7 @@
backward : pool2d_grad backward : pool2d_grad
- op : pool3d - op : pool3d
args : (Tensor x, int[] kernel_size, int[] strides = {1,1,1}, int[] paddings = {0,0,0}, bool ceil_mode = false, bool exclusive = true, str data_format = "NCDHW", str pooling_type = "", bool global_pooling = false, bool adaptive = false, str padding_algorithm = "EXPLICIT") args : (Tensor x, int[] kernel_size, int[] strides = {1,1,1}, int[] paddings = {0,0,0}, bool ceil_mode = false, bool exclusive = true, str data_format = "NCDHW", str pooling_type = "", bool global_pooling = false, bool adaptive = false, str padding_algorithm = "EXPLICIT", bool use_cudnn = false)
output : Tensor(out) output : Tensor(out)
infer_meta : infer_meta :
func : PoolInferMeta func : PoolInferMeta
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册