Searched refs:getCpuDevice (Results 1 – 8 of 8) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestFailingDriver.cpp | 83 DeviceManager::getCpuDevice(), in SetUp()
154 ASSERT_EQ(steps[1]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()
157 ASSERT_EQ(steps[3]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()
|
| H A D | TestPartitioning.cpp | 1174 devices.push_back(DeviceManager::getCpuDevice()); in makeDevices()
1621 ASSERT_EQ(planC.forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1916 step1, &model1, DeviceManager::getCpuDevice(), in TEST_F()
1971 ASSERT_EQ(cPNo.getExecutionPlan().forTest_simpleGetDevice(), DeviceManager::getCpuDevice()); in TEST_F()
1984 DeviceManager::getCpuDevice()); in TEST_F()
2227 const auto& cpuDeviceName = DeviceManager::getCpuDevice()->getName(); in TEST_F()
2246 const auto& cpuDeviceName = DeviceManager::getCpuDevice()->getName(); in TEST_F()
2424 DeviceManager::getCpuDevice()); in compileModelAndComparePlan()
3239 const auto& cpuDeviceName = DeviceManager::getCpuDevice()->getName(); in TEST_F()
3304 checkExecutionPlanSteps(plan, {DeviceManager::getCpuDevice()->getName()}); in testIfUnknownSize()
[all …]
|
| H A D | TestPartitioningRandom.cpp | 1112 devices.push_back(DeviceManager::getCpuDevice()); in TEST_P()
1157 DeviceManager::getCpuDevice()); in TEST_P()
1164 ASSERT_TRUE(plan.forTest_simpleGetDevice() != DeviceManager::getCpuDevice()); in TEST_P()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/ |
| H A D | Manager.h | 196 static std::shared_ptr<Device> getCpuDevice();
222 return reinterpret_cast<const Device*>(device) == getCpuDevice().get(); in forTest_isCpuDevice()
|
| H A D | CompilationBuilder.cpp | 100 mPlan.becomeSingleStep(DeviceManager::getCpuDevice(), mModel); in finish()
|
| H A D | ExecutionBuilder.cpp | 611 DeviceManager::getCpuDevice(), /*preparedModel=*/nullptr, in cpuFallbackFull()
1440 return mDevice == DeviceManager::getCpuDevice(); in isCpu()
1539 auto [n, preparedModel] = DeviceManager::getCpuDevice()->prepareModel(makeModel, preference, in computeOnCpuFallback()
|
| H A D | ExecutionPlan.cpp | 1924 return isSimple() && simple()->mDevice == DeviceManager::getCpuDevice(); in isSimpleCpu()
2508 std::find(devices.begin(), devices.end(), DeviceManager::getCpuDevice()); in findBestDeviceForEachOperation()
2523 const bool deviceIsPreferred = (device == DeviceManager::getCpuDevice() || in findBestDeviceForEachOperation()
2546 } else if (devices[bestChoice] == DeviceManager::getCpuDevice() && in findBestDeviceForEachOperation()
|
| H A D | Manager.cpp | 1224 std::shared_ptr<Device> DeviceManager::getCpuDevice() { in getCpuDevice() function in android::nn::DeviceManager
|