From 24e1c35c1f732d80711d9b8b98e9a829f944653b Mon Sep 17 00:00:00 2001 From: mengyuanli Date: Thu, 3 Sep 2020 20:42:57 +0800 Subject: [PATCH] fix buf of creating pooling --- mindspore/lite/src/ops/pooling.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/src/ops/pooling.cc b/mindspore/lite/src/ops/pooling.cc index 2b409622ca..9bac3a6030 100644 --- a/mindspore/lite/src/ops/pooling.cc +++ b/mindspore/lite/src/ops/pooling.cc @@ -147,7 +147,7 @@ int Pooling::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers auto val_offset = schema::CreatePooling(*fbb, attr->format(), attr->poolingMode(), attr->global(), attr->windowW(), attr->windowH(), 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); fbb->Finish(prim_offset); return RET_OK;