mirror of https://github.com/alibaba/MNN.git
36 lines
1.0 KiB
C++
36 lines
1.0 KiB
C++
//
|
|
// CPUSqueeze.cpp
|
|
// MNN
|
|
//
|
|
// Created by MNN on 2018/08/10.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#include "backend/cpu/CPUSqueeze.hpp"
|
|
#include "backend/cpu/CPUBackend.hpp"
|
|
|
|
namespace MNN {
|
|
|
|
CPUSqueeze::CPUSqueeze(Backend* b) : MNN::Execution(b) {
|
|
// nothing to do
|
|
}
|
|
|
|
ErrorCode CPUSqueeze::onExecute(const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs) {
|
|
auto& dstBuffer = outputs[0]->buffer();
|
|
auto& srcBuffer = inputs[0]->buffer();
|
|
memcpy(dstBuffer.host, srcBuffer.host, outputs[0]->size());
|
|
return NO_ERROR;
|
|
}
|
|
|
|
class CPUSqueezeCreator : public CPUBackend::Creator {
|
|
public:
|
|
virtual Execution* onCreate(const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs,
|
|
const MNN::Op* op, Backend* backend) const override {
|
|
return new CPUSqueeze(backend);
|
|
}
|
|
};
|
|
|
|
REGISTER_CPU_OP_CREATOR(CPUSqueezeCreator, OpType_Squeeze);
|
|
REGISTER_CPU_OP_CREATOR(CPUSqueezeCreator, OpType_Unsqueeze);
|
|
} // namespace MNN
|