未验证 提交 b582355e 编写于 作者: derwoo's avatar derwoo 提交者: GitHub

Fixing keys in SaveTmHardSwishOp (#903)

上级 7260bf4a
......@@ -1095,7 +1095,7 @@ tm_uoffset_t SaveTmHardSwishOp(void* const start_ptr, tm_uoffset_t* cur_pos, ir_
tm_param.beta = p->beta;
TM2_Operator tm_op;
SetTmOperator(&tm_op, TM2_OPTYPE_CAST, WriteTmObject(start_ptr, cur_pos, &tm_param, sizeof(TM2_HardSwishParam)));
SetTmOperator(&tm_op, TM2_OPTYPE_HARDSWISH, WriteTmObject(start_ptr, cur_pos, &tm_param, sizeof(TM2_HardSwishParam)));
return WriteTmObject(start_ptr, cur_pos, &tm_op, sizeof(TM2_Operator));
}
tm_uoffset_t SaveTmInterpOp(void* const start_ptr, tm_uoffset_t* cur_pos, ir_node_t* node)
......@@ -1570,4 +1570,4 @@ op_save_t SaveTmOpFunc(uint32_t op_type)
// fprintf(stderr, "Operator #%d not supported in tengine model yet\n",op_type);
return nullptr;
}
}
\ No newline at end of file
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册