未验证 提交 3f49817a 编写于 作者: Y yeliang2258 提交者: GitHub

fix mkldnn interpolate ops (#45008)

上级 35839aee
......@@ -105,7 +105,8 @@ class InterpolateMKLDNNKernel : public framework::OpKernel<T> {
}
}
}
if (scale[0] > 0.0f && scale[1] > 0.0f && scale[2] > 0.0f) {
if (scale.size() == 3 && scale[0] > 0.0f && scale[1] > 0.0f &&
scale[2] > 0.0f) {
int j = 0;
std::vector<int64_t> in_dhw_vec = phi::vectorize(in_dhw_dims);
std::transform(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册