Searched refs:getCpuDevice (Results 1 – 7 of 7) 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 | 1072 devices.push_back(DeviceManager::getCpuDevice()); in makeDevices()
1461 ASSERT_EQ(planC.forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1742 step1, &model1, DeviceManager::getCpuDevice(), in TEST_F()
1799 ASSERT_EQ(cPNo.getExecutionPlan().forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1814 DeviceManager::getCpuDevice()); in TEST_F()
2680 DeviceManager::getCpuDevice()->getName()); in TEST_F()
2731 ASSERT_EQ(plan.forTest_simpleGetDevice()->getName(), DeviceManager::getCpuDevice()->getName()); in TEST_F()
2751 ASSERT_EQ(plan.forTest_simpleGetDevice()->getName(), DeviceManager::getCpuDevice()->getName()); in TEST_F()
|
| 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 | 468 DeviceManager::getCpuDevice(), /*preparedModel=*/nullptr); in cpuFallbackFull()
1066 return mDevice == DeviceManager::getCpuDevice(); in isCpu()
1131 mDevice = DeviceManager::getCpuDevice(); in computeOnCpuFallback()
|
| D | Manager.cpp | 816 std::shared_ptr<Device> DeviceManager::getCpuDevice() { in getCpuDevice() function in android::nn::DeviceManager
|