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