未验证 提交 1b7f6ae9 编写于 作者: B baoachun 提交者: GitHub

add gelu pbtxt for conv+gelu mkldnn fuse pass (#38162)

上级 76514a1f
......@@ -253,6 +253,7 @@ void CpuPassStrategy::EnableMKLDNN() {
"conv_swish_mkldnn_fuse_pass", //
"conv_hard_swish_mkldnn_fuse_pass", //
"conv_hard_sigmoid_mkldnn_fuse_pass", //
// TODO(baoachun) fix int8 accuracy
"conv_gelu_mkldnn_fuse_pass",
"scale_matmul_fuse_pass", //
"reshape_transpose_matmul_mkldnn_fuse_pass", //
......
type: "gelu"
def {
inputs {
name: "X"
}
outputs {
name: "Out"
}
attrs {
name: "approximate"
type: BOOLEAN
}
}
extra {
attrs {
name: "use_mkldnn"
type: BOOLEAN
}
attrs {
name: "mkldnn_data_type"
type: STRING
}
attrs {
name: "use_cudnn"
type: BOOLEAN
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册