Method onCompute

source/geometry/GeometryTensorArray.cpp:486–526

class GeometryTensorArrayErase : public GeometryComputer {
public:
    virtual bool onCompute(const Op* op, const std::vector<Tensor*>& inputs, const std::vector<Tensor*>& outputs,
                           Context& context, CommandBuffer& res) const override {
        auto tensorArrayInput = inputs[2];
        auto inDes = TensorUtils::getDescribe(tensorArrayInput);
        if (inDes->tensorArrayAttr == nullptr) {
            MNN_ASSERT(false);
            return false;
        }
        auto output    = outputs[0];
        auto outputDes = TensorUtils::getDescribe(output);
        outputDes->memoryType = Tensor::InsideDescribe::MEMORY_VIRTUAL;

        int eraseIndex = inputs[1]->host<int32_t>()[0], oldSize = inDes->tensorArrayAttr->arraySize;
        eraseIndex += (eraseIndex < 0 ? oldSize: 0);
        auto eleSize = getElemSize(tensorArrayInput, eraseIndex);
        outputDes->regions.clear();
        if (eraseIndex > 0) {
            Tensor::InsideDescribe::Region reg;
            reg.origin = tensorArrayInput;
            reg.src.offset = 0;
            reg.src.stride[0] = reg.src.stride[1] = reg.src.stride[2] = 1;
            reg.dst.offset = 0;
            reg.dst.stride[0] = reg.dst.stride[1] = reg.dst.stride[2] = 1;
            reg.size[0] = eleSize.first;
            reg.size[1] = reg.size[2] = 1;
            outputDes->regions.push_back(reg);
        }
        if (eraseIndex < oldSize - 1) {
            int offset = eleSize.first + eleSize.second;
            Tensor::InsideDescribe::Region reg;
            reg.origin = tensorArrayInput;
            reg.src.offset = offset;
            reg.src.stride[0] = reg.src.stride[1] = reg.src.stride[2] = 1;
            reg.dst.offset = eleSize.first;
            reg.dst.stride[0] = reg.dst.stride[1] = reg.dst.stride[2] = 1;
            reg.size[0] = tensorArrayInput->elementSize() - offset;
            reg.size[1] = reg.size[2] = 1;
            outputDes->regions.push_back(reg);
        }
        return true;
    }
};

static void _create() {