未验证 提交 57e8cbec 编写于 作者: J jakpiase 提交者: GitHub

Fix for oneDNN conv op (#36284)

* fix for conv op

* Minor change
上级 f9591bb1
......@@ -217,8 +217,9 @@ class ConvMKLDNNHandlerT
const auto fwd_prop_kind = is_test ? mkldnn::prop_kind::forward_inference
: mkldnn::prop_kind::forward_training;
float sum_scale;
float sum_scale = 1.0f;
std::vector<float> output_shift_scale;
if (platform::is_int8<T>())
std::tie(sum_scale, output_shift_scale) = get_int8_scales(ctx);
const mkldnn::primitive_attr conv_attr = CreatePostOps(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册