|
|
|
@ -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);
|
|
|
|
|