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

fix(sdk): fix error caused by incorrect parsing u16/s16 inputs in lar

GitOrigin-RevId: 9486cf362d25b898b44e023d9fd2a49d7873dd96
上级 177dec94
......@@ -423,10 +423,13 @@ private:
}
const std::map<std::string, megdnn::DType> type_map = {
{"f4", dtype::Float32()},
{"f4", dtype::Float32()},
{"i4", dtype::Int32()},
{"i1", dtype::Int8()},
{"u1", dtype::Uint8()}};
{"u2", dtype::Uint16()},
{"i2", dtype::Int16()},
{"i1", dtype::Int8()},
{"u1", dtype::Uint8()},
};
megdnn::DType hv_type;
for (auto& item : type_map) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册