提交 6cf3da24 编写于 作者: 卢旭辉

Merge branch 'onnx' into 'master'

Fix ONNX padding and add transpose half

See merge request deep-computing/mace!1271
......@@ -65,6 +65,8 @@ class TransposeOp : public Operation {
void RegisterTranspose(OpRegistry *op_registry) {
MACE_REGISTER_OP(op_registry, "Transpose", TransposeOp,
DeviceType::CPU, float);
MACE_REGISTER_OP(op_registry, "Transpose", TransposeOp,
DeviceType::CPU, half);
MACE_REGISTER_BF16_OP(op_registry, "Transpose", TransposeOp,
DeviceType::CPU);
}
......
......@@ -1236,6 +1236,8 @@ class OnnxConverter(base_converter.ConverterInterface):
paddings_arg = op.arg.add()
paddings_arg.name = MaceKeyword.mace_paddings_str
paddings_value = node.attrs['pads']
paddings_value = np.asarray(paddings_value).reshape(
(2, -1)).transpose().reshape(-1).tolist()
paddings_arg.ints.extend(paddings_value)
if 'value' in node.attrs:
constant_value_arg = op.arg.add()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册