blob: 6b8167aeaac101d859ae215c1da8c918b5b18b27 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
//
// 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
|