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

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

上级 610a47dd
......@@ -1979,7 +1979,7 @@
pool2d_double_grad : GetPoolDoubleGradExpectedKernelType
extra :
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
backward : pool3d_grad
......@@ -1993,7 +1993,7 @@
pool3d : GetPoolExpectedKernelType
pool3d_grad : GetPoolExpectedKernelType
extra :
attrs : [bool use_mkldnn = false, bool use_cudnn = false]
attrs : [bool use_mkldnn = false]
- op : pow
backward : pow_grad, pow_double_grad, pow_triple_grad
......
......@@ -412,7 +412,7 @@
param : [peer, dtype, out_shape]
- 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)
infer_meta :
func : Pool2DInferMeta
......@@ -423,7 +423,7 @@
backward : pool2d_grad
- 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)
infer_meta :
func : PoolInferMeta
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册