Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1638 std::vector<int> bestDeviceForOperation(operationCount); in partitionTheWorkInternal() local
1640 findBestDeviceForEachOperation(preference, devices, &bestDeviceForOperation)); in partitionTheWorkInternal()
1651 std::adjacent_find(bestDeviceForOperation.begin(), bestDeviceForOperation.end(), in partitionTheWorkInternal()
1652 std::not_equal_to<int>()) == bestDeviceForOperation.end()) { in partitionTheWorkInternal()
1653 const int bestDeviceIndex = bestDeviceForOperation[0]; in partitionTheWorkInternal()
1673 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWorkInternal()
1992 std::vector<int>* bestDeviceForOperation) const { in findBestDeviceForEachOperation()
2051 (*bestDeviceForOperation)[operationIndex] = kControlFlowInterpreter; in findBestDeviceForEachOperation()
2056 (*bestDeviceForOperation)[operationIndex] = bestChoice; in findBestDeviceForEachOperation()
DModelBuilder.h150 std::vector<int>* bestDeviceForOperation) const;