chenjianping
5 years ago
parent
29070d60a1
commit
34f21226a8
@ -1,35 +0,0 @@
|
|||||||
/**
|
|
||||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "src/ops/prelu.h"
|
|
||||||
|
|
||||||
namespace mindspore {
|
|
||||||
namespace lite {
|
|
||||||
#ifdef PRIMITIVE_WRITEABLE
|
|
||||||
std::vector<float> Prelu::GetSlope() const { return this->primitive_->value.AsPrelu()->slope; }
|
|
||||||
|
|
||||||
void Prelu::SetSlope(const std::vector<float> &slope) { this->primitive_->value.AsPrelu()->slope = slope; }
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
std::vector<float> Prelu::GetSlope() const {
|
|
||||||
auto fb_vector = this->primitive_->value_as_Prelu()->slope();
|
|
||||||
return std::vector<float>(fb_vector->begin(), fb_vector->end());
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
} // namespace lite
|
|
||||||
} // namespace mindspore
|
|
||||||
@ -1,72 +0,0 @@
|
|||||||
/**
|
|
||||||
* Copyright 2019-2020 Huawei Technologies Co., Ltd
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef LITE_MINDSPORE_LITE_C_OPS_PRELU_H_
|
|
||||||
#define LITE_MINDSPORE_LITE_C_OPS_PRELU_H_
|
|
||||||
|
|
||||||
#include <vector>
|
|
||||||
#include <set>
|
|
||||||
#include <cmath>
|
|
||||||
#include <memory>
|
|
||||||
#include "ir/dtype/type_id.h"
|
|
||||||
#include "src/ops/activation.h"
|
|
||||||
|
|
||||||
namespace mindspore {
|
|
||||||
namespace lite {
|
|
||||||
class Prelu : public Activation {
|
|
||||||
public:
|
|
||||||
#ifdef PRIMITIVE_WRITEABLE
|
|
||||||
MS_DECLARE_PARENT(Prelu, PrimitiveC);
|
|
||||||
Prelu() = default;
|
|
||||||
explicit Prelu(schema::PrimitiveT *primitive) : Activation(primitive) {}
|
|
||||||
void SetSlope(const std::vector<float> &slope);
|
|
||||||
|
|
||||||
#else
|
|
||||||
explicit Prelu(schema::Primitive *primitive) : Activation(primitive) {}
|
|
||||||
|
|
||||||
schema::Primitive *Init(schema::Primitive *primitive) {
|
|
||||||
flatbuffers::FlatBufferBuilder fbb(1024);
|
|
||||||
|
|
||||||
auto attr = primitive->value_as_Prelu();
|
|
||||||
MS_ASSERT(attr != nullptr);
|
|
||||||
|
|
||||||
auto slope = std::make_unique<std::vector<float>>();
|
|
||||||
for (int i = 0; i < static_cast<int>(attr->slope()->size()); i++) {
|
|
||||||
slope->push_back(attr->slope()->data()[i]);
|
|
||||||
}
|
|
||||||
|
|
||||||
auto val_offset = schema::CreatePreluDirect(fbb, slope.release());
|
|
||||||
auto prim_offset = schema::CreatePrimitive(fbb, schema::PrimitiveType_Prelu, val_offset.o);
|
|
||||||
fbb.Finish(prim_offset);
|
|
||||||
|
|
||||||
auto buf = fbb.GetBufferPointer();
|
|
||||||
MS_ASSERT(buf != nullptr);
|
|
||||||
auto buf_bak = new char[fbb.GetSize()];
|
|
||||||
memcpy(buf_bak, buf, fbb.GetSize());
|
|
||||||
|
|
||||||
auto root = flatbuffers::GetRoot<schema::Primitive>(buf_bak);
|
|
||||||
auto prim = const_cast<schema::Primitive *>(root);
|
|
||||||
|
|
||||||
delete[] buf_bak;
|
|
||||||
fbb.Clear();
|
|
||||||
return prim;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
std::vector<float> GetSlope() const;
|
|
||||||
};
|
|
||||||
} // namespace lite
|
|
||||||
} // namespace mindspore
|
|
||||||
#endif // LITE_MINDSPORE_LITE_C_OPS_PRELU_H_
|
|
||||||
Loading…
Reference in new issue