driver/sample/SampleDriver.cpp

/*
 * Copyright (C) 2017 The Android Open Source Project
 *
 * 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.
 */

#define LOG_TAG "SampleDriver"

#include "SampleDriver.h"
#include "BinaryModel.h"
#include "CpuExecutor.h"

namespace android {
namespace nn {

int SampleRequest::execute(const std::vector<InputOutputInfo>& inputs,
                           const std::vector<InputOutputInfo>& outputs, IEvent** event) {
    // TODO: use a thread pool
    *event = new SampleEvent();
    CpuExecutor executor(mModel, inputs, outputs);
    return executor.run();
}

void SampleDriver::initialize([[maybe_unused]] Capabilities* capabilities) {
    // TODO Implement
}

void SampleDriver::getSupportedSubgraph([[maybe_unused]] void* graph,
                                        [[maybe_unused]] std::vector<bool>& canDo) {
    // TODO Implement
}

int SampleDriver::prepareRequest(const SerializedModel* modelBuffer, IRequest** request) {
    const auto model = BinaryModel::Create(modelBuffer->memory.data(), modelBuffer->memory.size());
    if (model == nullptr) {
        return ANEURALNETWORKS_BAD_DATA;
    }
    *request = new SampleRequest(model);
    return ANEURALNETWORKS_NO_ERROR;
}

Status SampleDriver::getStatus() {
    return AVAILABLE;
}

}  // namespace nn
}  // namespace android