未验证 提交 9034ca70 编写于 作者: Z zyfncg 提交者: GitHub

set use_cudnn=true for conv2d (#45587)

上级 3f280ab3
...@@ -30,7 +30,7 @@ ...@@ -30,7 +30,7 @@
- api : conv2d - api : conv2d
backward : conv2d_grad backward : conv2d_grad
extra : extra :
attrs : [bool use_cudnn = false, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false, attrs : [bool use_cudnn = true, bool fuse_relu_before_depthwise_conv = false, bool use_mkldnn = false,
bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false, bool use_quantizer = false, str mkldnn_data_type = "float32", bool fuse_relu = false,
str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false, str fuse_activation = "", float fuse_alpha = 0.0f, float fuse_beta = 0.0f, bool use_addto = false,
bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f, bool fuse_residual_connection = false, float Scale_in = 1.0f, float Scale_out = 1.0f,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册