Searched refs:getCpuDevice (Results 1 – 8 of 8) sorted by relevance
| /frameworks/ml/nn/runtime/test/ |
| D | TestFailingDriver.cpp | 84 DeviceManager::getCpuDevice(), in SetUp()
155 ASSERT_EQ(steps[1]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()
158 ASSERT_EQ(steps[3]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()
|
| D | TestPartitioning.cpp | 1082 devices.push_back(DeviceManager::getCpuDevice()); in makeDevices()
1471 ASSERT_EQ(planC.forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1752 step1, &model1, DeviceManager::getCpuDevice(), in TEST_F()
1808 ASSERT_EQ(cPNo.getExecutionPlan().forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1823 DeviceManager::getCpuDevice()); in TEST_F()
2694 DeviceManager::getCpuDevice()->getName()); in TEST_F()
2753 ASSERT_EQ(plan.forTest_simpleGetDevice()->getName(), DeviceManager::getCpuDevice()->getName()); in testIfUnknownSize()
2796 ASSERT_EQ(plan.forTest_simpleGetDevice()->getName(), DeviceManager::getCpuDevice()->getName()); in testWhileUnknownSize()
|
| D | TestPartitioningRandom.cpp | 1158 devices.push_back(DeviceManager::getCpuDevice()); in TEST_P()
1178 DeviceManager::getCpuDevice()); in TEST_P()
|
| /frameworks/ml/nn/runtime/ |
| D | Manager.h | 158 static std::shared_ptr<Device> getCpuDevice();
188 return reinterpret_cast<const Device*>(device) == getCpuDevice().get(); in forTest_isCpuDevice()
|
| D | CompilationBuilder.cpp | 98 mPlan.becomeSingleStep(DeviceManager::getCpuDevice(), mModel); in finish()
|
| D | ExecutionBuilder.cpp | 466 DeviceManager::getCpuDevice(), /*preparedModel=*/nullptr); in cpuFallbackFull()
1064 return mDevice == DeviceManager::getCpuDevice(); in isCpu()
1129 mDevice = DeviceManager::getCpuDevice(); in computeOnCpuFallback()
|
| D | ExecutionPlan.cpp | 1500 return isSimple() && simple()->mDevice == DeviceManager::getCpuDevice(); in isSimpleCpu()
2013 std::find(devices.begin(), devices.end(), DeviceManager::getCpuDevice()); in findBestDeviceForEachOperation()
2027 (perfVal == bestPerfVal && device == DeviceManager::getCpuDevice())) { in findBestDeviceForEachOperation()
2046 } else if (devices[bestChoice] == DeviceManager::getCpuDevice() && in findBestDeviceForEachOperation()
|
| D | Manager.cpp | 815 std::shared_ptr<Device> DeviceManager::getCpuDevice() { in getCpuDevice() function in android::nn::DeviceManager
|