From b3f8df59d49fff58504f9e1adb6be12575e4ec2b Mon Sep 17 00:00:00 2001 From: Haihao Shen Date: Fri, 9 Nov 2018 12:37:27 +0800 Subject: [PATCH] Add conv attribute --- paddle/fluid/operators/conv_op.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/fluid/operators/conv_op.cc b/paddle/fluid/operators/conv_op.cc index 9a1e8614782..424f07980f3 100644 --- a/paddle/fluid/operators/conv_op.cc +++ b/paddle/fluid/operators/conv_op.cc @@ -189,6 +189,8 @@ void Conv2DOpMaker::Make() { "whenever convolution output is as an input to residual " "connection.") .SetDefault(false); + AddAttr("fuse_fp32_output", "(bool, default false) Force INT8 kernel output FP32, only used in mkldnn kernel") + .SetDefault(false); AddAttr( "data_format", "(string, default NCHW) Only used in " -- GitLab