Browse Source

remove adder has bias

tags/v1.2.0-rc1
zhaozhenlong 5 years ago
parent
commit
e2992b10f8
2 changed files with 4 additions and 9 deletions
  1. +4
    -7
      mindspore/lite/src/ops/adder.cc
  2. +0
    -2
      mindspore/lite/src/ops/adder.h

+ 4
- 7
mindspore/lite/src/ops/adder.cc View File

@@ -46,7 +46,6 @@ int Adder::GetPadLeft() const { return this->primitive_->value.AsAdder()->padLef
int Adder::GetPadRight() const { return this->primitive_->value.AsAdder()->padRight; }
int Adder::GetDilateW() const { return this->primitive_->value.AsAdder()->dilateW; }
int Adder::GetDilateH() const { return this->primitive_->value.AsAdder()->dilateH; }
bool Adder::GetHasBias() const { return this->primitive_->value.AsAdder()->hasBias; }
int Adder::GetActivationType() const { return this->primitive_->value.AsAdder()->activationType; }

void Adder::SetFormat(int format) { this->primitive_->value.AsAdder()->format = (schema::Format)format; }
@@ -64,7 +63,6 @@ void Adder::SetPadLeft(int pad_left) { this->primitive_->value.AsAdder()->padLef
void Adder::SetPadRight(int pad_right) { this->primitive_->value.AsAdder()->padRight = pad_right; }
void Adder::SetDilateW(int dilate_w) { this->primitive_->value.AsAdder()->dilateW = dilate_w; }
void Adder::SetDilateH(int dilate_h) { this->primitive_->value.AsAdder()->dilateH = dilate_h; }
void Adder::SetHasBias(bool has_bias) { this->primitive_->value.AsAdder()->hasBias = has_bias; }
void Adder::SetActivationType(int activation_type) {
this->primitive_->value.AsAdder()->activationType = (schema::ActivationType)activation_type;
}
@@ -154,10 +152,10 @@ int Adder::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffers::
return RET_ERROR;
}

auto val_offset = schema::CreateAdder(
*fbb, attr->format(), attr->group(), attr->channelIn(), attr->channelOut(), attr->kernelW(), attr->kernelH(),
attr->strideW(), attr->strideH(), attr->padMode(), attr->padUp(), attr->padDown(), attr->padLeft(),
attr->padRight(), attr->dilateW(), attr->dilateH(), attr->hasBias(), attr->activationType());
auto val_offset = schema::CreateAdder(*fbb, attr->format(), attr->group(), attr->channelIn(), attr->channelOut(),
attr->kernelW(), attr->kernelH(), attr->strideW(), attr->strideH(),
attr->padMode(), attr->padUp(), attr->padDown(), attr->padLeft(),
attr->padRight(), attr->dilateW(), attr->dilateH(), attr->activationType());
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Adder, val_offset.o);
fbb->Finish(prim_offset);
return RET_OK;
@@ -178,7 +176,6 @@ int Adder::GetPadLeft() const { return this->primitive_->value_as_Adder()->padLe
int Adder::GetPadRight() const { return this->primitive_->value_as_Adder()->padRight(); }
int Adder::GetDilateW() const { return this->primitive_->value_as_Adder()->dilateW(); }
int Adder::GetDilateH() const { return this->primitive_->value_as_Adder()->dilateH(); }
bool Adder::GetHasBias() const { return this->primitive_->value_as_Adder()->hasBias(); }
int Adder::GetActivationType() const { return this->primitive_->value_as_Adder()->activationType(); }

PrimitiveC *AdderCreator(const schema::Primitive *primitive) { return PrimitiveC::NewPrimitiveC<Adder>(primitive); }


+ 0
- 2
mindspore/lite/src/ops/adder.h View File

@@ -49,7 +49,6 @@ class Adder : public Conv2D {
void SetPadRight(int pad_right);
void SetDilateW(int dilate_w);
void SetDilateH(int dilate_h);
void SetHasBias(bool has_bias);
void SetActivationType(int activation_type);

private:
@@ -74,7 +73,6 @@ class Adder : public Conv2D {
int GetPadRight() const;
int GetDilateW() const;
int GetDilateH() const;
bool GetHasBias() const;
int GetActivationType() const;
};
} // namespace lite


Loading…
Cancel
Save