提交 de084f92 编写于 作者: M Megvii Engine Team

fix(src/atlas): remove the limitation that the input of the om model must be 4-dimensional

GitOrigin-RevId: ab3fce058e8ce1fa47e05142170b06a33bdcbf95
上级 c2b9d542
...@@ -40,9 +40,6 @@ TensorShape acl_shape_to_mgb_shape_for_input( ...@@ -40,9 +40,6 @@ TensorShape acl_shape_to_mgb_shape_for_input(
MGB_MARK_USED_VAR(aipp_input_fmt); MGB_MARK_USED_VAR(aipp_input_fmt);
TensorShape ret; TensorShape ret;
ret.ndim = acl_shape.dimCount; ret.ndim = acl_shape.dimCount;
mgb_assert(
ret.ndim == 4, "Unexpected ndim form aclmdlIODims expected 4, but got %zu",
ret.ndim);
for (size_t i = 0; i < ret.ndim; ++i) { for (size_t i = 0; i < ret.ndim; ++i) {
ret[i] = acl_shape.dims[i]; ret[i] = acl_shape.dims[i];
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册