Lines Matching refs:devices
847 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() argument
851 ->partitionTheWork(devices, static_cast<uint32_t>(preference), in partitionTheWork()
943 const std::vector<std::shared_ptr<Device>>& devices) { in PartitioningCompilation() argument
946 int result = m->createCompilation(&c, devices); in PartitioningCompilation()
1129 std::vector<std::shared_ptr<Device>> devices; in makeDevices() local
1173 devices.push_back(std::move(driverDevice)); in makeDevices()
1175 devices.push_back(DeviceManager::getCpuDevice()); in makeDevices()
1176 return devices; in makeDevices()
1827 const auto devices = makeDevices({{"V1_0", 0.6, HalVersion::V1_0, ~0U}, in TEST_F() local
1832 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
1849 const auto devices = makeDevices({{"1", 0.5, 1 << kDevOp}}); in TEST_F() local
1872 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
1893 compare(step0, &model0, devices[0], in TEST_F()
1940 step2, &model2, devices[0], RemapVectorType{{opnd6, m2Opnd6}}, // modelInputs in TEST_F()
1961 const auto devices = makeDevices({{"hw", 0.5, (1 << 0)}}); in TEST_F() local
1967 PartitioningCompilation cPNo(&model, devices); in TEST_F()
1978 PartitioningCompilation cPWithFallback(&model, devices); in TEST_F()
1989 PartitioningCompilation cPWithoutFallback(&model, devices); in TEST_F()
2014 const auto devices = makeDevices({{"0", 0.5, 1 << 0}, {"1", 0.5, 1 << 1}}); in TEST_F() local
2016 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
2033 compare(steps[0], &model0, devices[0], in TEST_F()
2051 compare(steps[1], &model1, devices[1], RemapVectorType{}, // modelInputs in TEST_F()
2100 const auto devices = makeDevices({// Best choice for non-relaxed model. in TEST_F() local
2105 auto TrivialTest = [&devices](bool doRelax, const char* expectDevice) { in TEST_F()
2120 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
2166 const auto devices = in TEST_F() local
2174 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
2186 const auto devices = in TEST_F() local
2194 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
2225 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2226 PartitioningCompilation compilation(&model, devices); in TEST_F()
2244 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2245 PartitioningCompilation compilation(&model, devices); in TEST_F()
2345 auto devices = makeDevices({{"pad", in compileModelAndComparePlan() local
2358 mCompilation = PartitioningCompilation(&mModel.value(), devices); in compileModelAndComparePlan()
2386 stepsA[0], &modelA0, devices[0], in compileModelAndComparePlan()
2409 compare(stepsA[1], &modelA1, devices[1], RemapVectorType{}, // modelInputs in compileModelAndComparePlan()
2419 mCompilation = PartitioningCompilation(&mModel.value(), devices); in compileModelAndComparePlan()
2643 const std::vector<std::shared_ptr<Device>>& devices, in getTransformedCacheTokenSingle() argument
2649 PartitioningCompilation compilation(&model, devices); in getTransformedCacheTokenSingle()
2696 const std::vector<std::shared_ptr<Device>>& devices, in getTransformedCacheToken() argument
2701 getTransformedCacheTokenSingle(model, devices, deviceName, tokenIn, preference, priority, in getTransformedCacheToken()
2709 getTransformedCacheTokenSingle(model, devices, deviceName, tokenIn, preference, in getTransformedCacheToken()
2907 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2910 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2914 getTransformedCacheToken(model, devices, "deviceB", tokenIn, in TEST_F()
2971 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2975 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2978 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2981 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2994 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2998 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
3001 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
3004 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
3017 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
3022 getTransformedCacheToken(model, devices, "deviceA", tokenIn1, in TEST_F()
3025 getTransformedCacheToken(model, devices, "deviceA", tokenIn2, in TEST_F()
3066 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}}); in TEST_F() local
3070 getTransformedCacheToken(main, devices, "deviceA", tokenIn, in TEST_F()
3073 getTransformedCacheToken(main, devices, "deviceA", tokenIn, in TEST_F()
3221 const auto devices = makeDevices({{"V1_0", 0.9, HalVersion::V1_0, ~0U}}); in TEST_F() local
3224 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
3235 const auto devices = makeDevices({{"V1_0", 0.9, HalVersion::V1_0, ~0U}}); in TEST_F() local
3238 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
3249 const auto devices = makeDevices({{"ALL", in TEST_F() local
3257 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
3267 const auto devices = makeDevices({{"ALL", in TEST_F() local
3275 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in TEST_F()
3294 const auto devices = makeDevices({{"ALL", in testIfUnknownSize() local
3302 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in testIfUnknownSize()
3337 const auto devices = makeDevices({{"ALL", in testWhileUnknownSize() local
3345 ASSERT_EQ(models[0]->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in testWhileUnknownSize()
3392 void finishAndPartitionModelForDevices(const std::vector<std::shared_ptr<Device>>& devices) { in finishAndPartitionModelForDevices() argument
3395 ASSERT_EQ(mModel->partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, in finishAndPartitionModelForDevices()
3453 const auto devices = makeDevices( in TEST_F() local
3455 finishAndPartitionModelForDevices(devices); in TEST_F()
3504 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
3505 finishAndPartitionModelForDevices(devices); in TEST_F()
3562 const auto devices = makeDevices({{"deviceA", in TEST_F() local
3569 finishAndPartitionModelForDevices(devices); in TEST_F()
3653 const auto devices = makeDevices({{"deviceA", in TEST_F() local
3661 finishAndPartitionModelForDevices(devices); in TEST_F()