Browse Source

!5726 [MS][LITE][Develop]fix bug of creating pooling op

Merge pull request !5726 from mengyuanli/move_allocator
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
9af15bb8a7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/ops/pooling.cc

+ 1
- 1
mindspore/lite/src/ops/pooling.cc View File

@@ -147,7 +147,7 @@ int Pooling::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers
auto val_offset = auto val_offset =
schema::CreatePooling(*fbb, attr->format(), attr->poolingMode(), attr->global(), attr->windowW(), attr->windowH(), schema::CreatePooling(*fbb, attr->format(), attr->poolingMode(), attr->global(), attr->windowW(), attr->windowH(),
attr->strideW(), attr->strideH(), attr->padMode(), attr->padUp(), attr->padDown(), attr->strideW(), attr->strideH(), attr->padMode(), attr->padUp(), attr->padDown(),
attr->padLeft(), attr->padRight(), attr->roundMode());
attr->padLeft(), attr->padRight(), attr->roundMode(), attr->activationType());
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Pooling, val_offset.o); auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Pooling, val_offset.o);
fbb->Finish(prim_offset); fbb->Finish(prim_offset);
return RET_OK; return RET_OK;


Loading…
Cancel
Save