Cross Reference: /frameworks/ml/nn/runtime/ExecutionPlan.cpp

Lines Matching defs:operationIndex
87     void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
99     for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) {
100         const Operation& operation = operations[operationIndex];
108                         std::pair<uint32_t, uint32_t>(operandIndex, operationIndex));
112             cb(operationIndex);
114         mUnknownInputCount[operationIndex] = count;
118 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) {
120     const Operation& operation = mModel->getOperations()[operationIndex];
222 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) {
223     const Operation& operation = fromModel.getOperation(operationIndex);
705     auto enqueueOnAppropriateDevice = [&](uint32_t operationIndex) {
706         int deviceIndex = bestDeviceForOperation[operationIndex];
707         perDeviceQueue[deviceIndex].push(operationIndex);
708         VLOG(COMPILATION) << "enqueueOnAppropriateDevice " << operationIndex << " onto "
744             uint32_t operationIndex = queue.front();
746             step->addOperation(operationIndex, *this);
747             tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice);
763                                                  uint32_t operationIndex) const {
764     const Operation& operation = getOperation(operationIndex);
799     bool check(size_t operationIndex) const { return mSupportsOperationByIndex[operationIndex]; }
831     for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) {
835             if (canDo[deviceIndex].check(operationIndex)) {
837                 const PerformanceInfo perf = getPerformanceInfo(device, operationIndex);
850         (*bestDeviceForOperation)[operationIndex] =
853                           << toString(getOperation(operationIndex).type)
855                           << (*bestDeviceForOperation)[operationIndex];