fix code review

This commit is contained in:
sunsuodong 2020-12-07 20:51:08 +08:00
parent c8dec08070
commit 796df828ed
1 changed files with 2 additions and 2 deletions

View File

@ -23,9 +23,9 @@
namespace mindspore {
namespace lite {
OpParameter *PopulateSpaceToBatchNDParameter(const mindspore::lite::PrimitiveC *primitive) {
auto *space_batch_param_nd = new (std::nothrow) SpaceToBatchParameter();
auto *space_batch_param_nd = reinterpret_cast<SpaceToBatchParameter *>(malloc(sizeof(SpaceToBatchParameter)));
if (space_batch_param_nd == nullptr) {
MS_LOG(ERROR) << "new SpaceToBatchParameter failed.";
MS_LOG(ERROR) << "malloc SpaceToBatchParameter failed.";
return nullptr;
}