Lines Matching refs:devices
577 int partitionTheWork(const std::vector<std::shared_ptr<Device>>& devices, in partitionTheWork() argument
580 devices, static_cast<uint32_t>(preference), plan); in partitionTheWork()
640 const std::vector<std::shared_ptr<Device>>& devices) { in PartitioningCompilation() argument
643 int result = m->createCompilation(&c, devices); in PartitioningCompilation()
792 std::vector<std::shared_ptr<Device>> devices; in makeDevices() local
818 devices.push_back(device); in makeDevices()
820 devices.push_back(DeviceManager::getCpuDevice()); in makeDevices()
821 return devices; in makeDevices()
1378 const auto devices = makeDevices({{"V1_0", 0.6, HalVersion::V1_0, ~0U}, in TEST_F() local
1382 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1397 const auto devices = makeDevices({{"1", 0.5, 1 << kDevOp}}); in TEST_F() local
1420 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1439 compare(step0, &model0, devices[0], in TEST_F()
1484 step2, &model2, devices[0], RemapVectorType{{opnd6, m2Opnd6}}, // modelInputs in TEST_F()
1509 const auto devices = makeDevices({{"hw", 0.5, (1 << 0)}}); in TEST_F() local
1515 PartitioningCompilation cPNo(&model, devices); in TEST_F()
1528 PartitioningCompilation cPWithFallback(&model, devices); in TEST_F()
1537 PartitioningCompilation cPWithoutFallback(&model, devices); in TEST_F()
1560 const auto devices = makeDevices({{"0", 0.5, 1 << 0}, {"1", 0.5, 1 << 1}}); in TEST_F() local
1562 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1577 compare(steps[0], &model0, devices[0], in TEST_F()
1594 compare(steps[1], &model1, devices[1], RemapVectorType{}, // modelInputs in TEST_F()
1642 const auto devices = makeDevices({// Best choice for non-relaxed model. in TEST_F() local
1647 auto TrivialTest = [&devices](bool doRelax, const char* expectDevice) { in TEST_F()
1662 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1702 const auto devices = in TEST_F() local
1710 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1720 const auto devices = in TEST_F() local
1728 ASSERT_EQ(model.partitionTheWork(devices, ExecutePreference::PREFER_LOW_POWER, &plan), in TEST_F()
1778 const std::vector<std::shared_ptr<Device>>& devices, in getTransformedCacheTokenSingle() argument
1783 PartitioningCompilation compilation(&model, devices); in getTransformedCacheTokenSingle()
1825 const std::vector<std::shared_ptr<Device>>& devices, in getTransformedCacheToken() argument
1828 getTransformedCacheTokenSingle(model, devices, deviceName, tokenIn, preference, tokenOut); in getTransformedCacheToken()
1835 getTransformedCacheTokenSingle(model, devices, deviceName, tokenIn, preference, &token); in getTransformedCacheToken()
1955 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
1958 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
1961 getTransformedCacheToken(model, devices, "deviceB", tokenIn, in TEST_F()
2013 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2017 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2019 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2021 getTransformedCacheToken(model, devices, "deviceA", tokenIn, in TEST_F()
2033 const auto devices = makeDevices({{"deviceA", 0.8, ~0U}, {"deviceB", 0.5, 1 << 1}}); in TEST_F() local
2038 getTransformedCacheToken(model, devices, "deviceA", tokenIn1, in TEST_F()
2040 getTransformedCacheToken(model, devices, "deviceA", tokenIn2, in TEST_F()