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

Fix for oneDNN conv op (#36284)

* fix for conv op

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