Home
last modified time | relevance | path

Searched refs:bestDeviceForOperation (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp977 std::vector<int> bestDeviceForOperation(operationCount); in partitionTheWork() local
979 findBestDeviceForEachOperation(preference, devices, &bestDeviceForOperation)); in partitionTheWork()
982 if (std::adjacent_find(bestDeviceForOperation.begin(), bestDeviceForOperation.end(), in partitionTheWork()
983 std::not_equal_to<int>()) == bestDeviceForOperation.end()) { in partitionTheWork()
984 const int bestDeviceIndex = bestDeviceForOperation[0]; in partitionTheWork()
998 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWork()
1557 std::vector<int>* bestDeviceForOperation) const { in findBestDeviceForEachOperation()
1599 (*bestDeviceForOperation)[operationIndex] = bestChoice; in findBestDeviceForEachOperation()
DModelBuilder.h111 std::vector<int>* bestDeviceForOperation) const;