|
|
|
@@ -23,10 +23,6 @@ |
|
|
|
namespace mindspore { |
|
|
|
namespace lite { |
|
|
|
#ifdef PRIMITIVE_WRITEABLE |
|
|
|
int AddN::GetN() const { return this->primitive_->value.AsAddN()->N; } |
|
|
|
|
|
|
|
void AddN::SetN(int n) { this->primitive_->value.AsAddN()->N = n; } |
|
|
|
|
|
|
|
int AddN::UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) { |
|
|
|
if (this->primitive_ == nullptr) { |
|
|
|
this->primitive_ = new (std::nothrow) schema::PrimitiveT; |
|
|
|
@@ -54,17 +50,11 @@ int AddN::UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &input |
|
|
|
int AddN::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::FlatBufferBuilder *fbb) { |
|
|
|
MS_ASSERT(nullptr != primitive); |
|
|
|
MS_ASSERT(nullptr != fbb); |
|
|
|
auto attr = primitive->value_as_AddN(); |
|
|
|
if (attr == nullptr) { |
|
|
|
MS_LOG(ERROR) << "value_as_AddN return nullptr"; |
|
|
|
return RET_ERROR; |
|
|
|
} |
|
|
|
auto val_offset = schema::CreateAddN(*fbb, attr->N()); |
|
|
|
auto val_offset = schema::CreateAddN(*fbb, 0); |
|
|
|
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_AddN, val_offset.o); |
|
|
|
fbb->Finish(prim_offset); |
|
|
|
return RET_OK; |
|
|
|
} |
|
|
|
int AddN::GetN() const { return this->primitive_->value_as_AddN()->N(); } |
|
|
|
|
|
|
|
PrimitiveC *AddNCreator(const schema::Primitive *primitive) { return PrimitiveC::NewPrimitiveC<AddN>(primitive); } |
|
|
|
Registry AddNRegistry(schema::PrimitiveType_AddN, AddNCreator); |
|
|
|
|