Home
last modified time | relevance | path

Searched refs:DeviceManager (Results 1 – 9 of 9) sorted by relevance

/frameworks/ml/nn/runtime/
DManager.cpp128 DeviceManager* DeviceManager::get() { in get()
129 static DeviceManager manager; in get()
133 void DeviceManager::findAvailableDevices() { in findAvailableDevices()
156 void DeviceManager::registerDevice(const char* name, const sp<V1_0::IDevice>& device) { in registerDevice()
163 DeviceManager::DeviceManager() { in DeviceManager() function in android::nn::DeviceManager
DCompilationBuilder.cpp31 mModel(model), mPartitioning(DeviceManager::get()->getPartitioning()) { in CompilationBuilder()
37 return finish(DeviceManager::get()->getDrivers()); in finish()
64 if (!DeviceManager::partitioningAllowsFallback(mPartitioning)) { in finish()
DManager.h68 class DeviceManager {
96 static DeviceManager* get();
100 DeviceManager();
DExecutionBuilder.cpp356 const bool allowFallback = DeviceManager::partitioningAllowsFallback(mPartitioning); in startCompute()
391 const std::vector<std::shared_ptr<Device>>& devices = DeviceManager::get()->getDrivers(); in startCompute()
/frameworks/ml/nn/runtime/test/
DTestExecution.cpp38 using DeviceManager = nn::DeviceManager; typedef
151 builder()->setPartitioning(DeviceManager::kPartitioningWithoutFallback); in TestCompilation()
DTestMain.cpp42 android::nn::DeviceManager::get()->setUseCpuOnly(true); in main()
DTestPartitioningRandom.cpp98 using DeviceManager = nn::DeviceManager; typedef
915 ASSERT_EQ(c.setPartitioning(DeviceManager::kPartitioningNo), Result::NO_ERROR); in TEST_P()
951 ASSERT_EQ(cNoFallback.setPartitioning(DeviceManager::kPartitioningWithoutFallback), in TEST_P()
956 ASSERT_EQ(cWithFallback.setPartitioning(DeviceManager::kPartitioningWithFallback), in TEST_P()
DTestPartitioning.cpp127 using DeviceManager = ::android::nn::DeviceManager; typedef
998 ASSERT_EQ(cPNo.setPartitioning(DeviceManager::kPartitioningNo), Result::NO_ERROR); in TEST_F()
1008 …ASSERT_EQ(cPWithFallback.setPartitioning(DeviceManager::kPartitioningWithFallback), Result::NO_ERR… in TEST_F()
1015 …ASSERT_EQ(cPWithoutFallback.setPartitioning(DeviceManager::kPartitioningWithoutFallback), Result::… in TEST_F()
DTestMemoryInternal.cpp228 android::nn::DeviceManager::get()->setUseCpuOnly(true); in TEST_F()