|
|
|
@ -24,10 +24,6 @@
|
|
|
|
|
namespace mindspore {
|
|
|
|
|
namespace lite {
|
|
|
|
|
#ifdef PRIMITIVE_WRITEABLE
|
|
|
|
|
std::vector<int> BiasAdd::GetAxis() const { return this->primitive_->value.AsBiasAdd()->axis; }
|
|
|
|
|
|
|
|
|
|
void BiasAdd::SetAxis(const std::vector<int> &axis) { this->primitive_->value.AsBiasAdd()->axis = axis; }
|
|
|
|
|
|
|
|
|
|
int BiasAdd::UnPackAttr(const Primitive &prim, const std::vector<AnfNodePtr> &inputs) {
|
|
|
|
|
if (this->primitive_ == nullptr) {
|
|
|
|
|
this->primitive_ = new (std::nothrow) schema::PrimitiveT;
|
|
|
|
@ -67,21 +63,11 @@ int BiasAdd::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers
|
|
|
|
|
MS_LOG(ERROR) << "value_as_BiasAdd return nullptr";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
std::vector<int32_t> axis;
|
|
|
|
|
if (attr->axis() != nullptr) {
|
|
|
|
|
for (int i = 0; i < static_cast<int>(attr->axis()->size()); i++) {
|
|
|
|
|
axis.push_back(attr->axis()->data()[i]);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
auto val_offset = schema::CreateBiasAddDirect(*fbb, &axis);
|
|
|
|
|
auto val_offset = schema::CreateBiasAddDirect(*fbb);
|
|
|
|
|
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_BiasAdd, val_offset.o);
|
|
|
|
|
fbb->Finish(prim_offset);
|
|
|
|
|
return RET_OK;
|
|
|
|
|
}
|
|
|
|
|
std::vector<int> BiasAdd::GetAxis() const {
|
|
|
|
|
auto fb_vector = this->primitive_->value_as_BiasAdd()->axis();
|
|
|
|
|
return std::vector<int>(fb_vector->begin(), fb_vector->end());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
PrimitiveC *BiasAddCreator(const schema::Primitive *primitive) { return PrimitiveC::NewPrimitiveC<BiasAdd>(primitive); }
|
|
|
|
|
Registry BiasAddRegistry(schema::PrimitiveType_BiasAdd, BiasAddCreator);
|
|
|
|
|