MNN/source/backend/cpu/CPUSetDiff1D.cpp

//
//  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