chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
//
|
||||
// CPUSetDiff1D.cpp
|
||||
// MNN
|
||||
//
|
||||
// Created by MNN on 2019/6/11.
|
||||
// Copyright © 2018 Alibaba. All rights reserved.
|
||||
//
|
||||
|
||||
#include "backend/cpu/CPUSetDiff1D.hpp"
|
||||
namespace MNN {
|
||||
ErrorCode CPUSetDiff1D::onExecute(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) {
|
||||
auto input = inputs[0];
|
||||
auto remove = inputs[1];
|
||||
if (input->getType().code != halide_type_int || remove->getType().code != halide_type_int) {
|
||||
return NOT_SUPPORT;
|
||||
}
|
||||
auto output = outputs[0];
|
||||
auto outputPtr = output->host<int32_t>();
|
||||
auto inputPtr = input->host<int32_t>();
|
||||
auto removePtr = remove->host<int32_t>();
|
||||
auto removeSize = remove->elementSize();
|
||||
auto inputSize = input->elementSize();
|
||||
int outputSize = 0;
|
||||
for (int i = 0; i < inputSize; ++i) {
|
||||
auto value = inputPtr[i];
|
||||
bool valid = true;
|
||||
for (int j = 0; j < removeSize; ++j) {
|
||||
if (value == removePtr[j]) {
|
||||
valid = false;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (valid) {
|
||||
outputPtr[outputSize++] = value;
|
||||
}
|
||||
}
|
||||
output->setLength(0, outputSize);
|
||||
return NO_ERROR;
|
||||
}
|
||||
class CPUSetDiff1DCreator : public CPUBackend::Creator {
|
||||
public:
|
||||
virtual Execution *onCreate(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs,
|
||||
const MNN::Op *op, Backend *backend) const {
|
||||
return new CPUSetDiff1D(backend);
|
||||
}
|
||||
};
|
||||
|
||||
REGISTER_CPU_OP_CREATOR(CPUSetDiff1DCreator, OpType_SetDiff1D);
|
||||
|
||||
}; // namespace MNN
|
||||
Reference in New Issue
Block a user