//
// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "canonical/ArmnnDriver.hpp"

#include <nnapi/IDevice.h>

namespace android::nn
{

std::vector<SharedDevice> getDevices()
{
    std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc, armnn::Compute::CpuRef, armnn::Compute::GpuAcc };
    return { std::make_shared<armnn_driver::ArmnnDriver>(DriverOptions(backends)) };
}

}  // namespace android::nn