|
|
|
@ -27,10 +27,8 @@ namespace mindspore {
|
|
|
|
|
namespace lite {
|
|
|
|
|
#ifdef PRIMITIVE_WRITEABLE
|
|
|
|
|
int Concat::GetAxis() const { return this->primitive_->value.AsConcat()->axis; }
|
|
|
|
|
int Concat::GetN() const { return this->primitive_->value.AsConcat()->n; }
|
|
|
|
|
|
|
|
|
|
void Concat::SetAxis(int axis) { this->primitive_->value.AsConcat()->axis = axis; }
|
|
|
|
|
void Concat::SetN(int n) { this->primitive_->value.AsConcat()->n = n; }
|
|
|
|
|
|
|
|
|
|
int Concat::UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) {
|
|
|
|
|
if (this->primitive_ == nullptr) {
|
|
|
|
@ -71,13 +69,12 @@ int Concat::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers:
|
|
|
|
|
MS_LOG(ERROR) << "value_as_Concat return nullptr";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
auto val_offset = schema::CreateConcat(*fbb, attr->axis(), attr->n());
|
|
|
|
|
auto val_offset = schema::CreateConcat(*fbb, attr->axis());
|
|
|
|
|
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Concat, val_offset.o);
|
|
|
|
|
fbb->Finish(prim_offset);
|
|
|
|
|
return RET_OK;
|
|
|
|
|
}
|
|
|
|
|
int Concat::GetAxis() const { return this->primitive_->value_as_Concat()->axis(); }
|
|
|
|
|
int Concat::GetN() const { return this->primitive_->value_as_Concat()->n(); }
|
|
|
|
|
|
|
|
|
|
PrimitiveC *ConcatCreator(const schema::Primitive *primitive) { return PrimitiveC::NewPrimitiveC<Concat>(primitive); }
|
|
|
|
|
Registry ConcatRegistry(schema::PrimitiveType_Concat, ConcatCreator);
|
|
|
|
|