Lines Matching refs:operationIndex
86 void markProcessed(uint32_t operationIndex, OperationReadyCallback cb);
98 for (uint32_t operationIndex = 0; operationIndex < operations.size(); operationIndex++) { in OperandTracker() local
99 const Operation& operation = operations[operationIndex]; in OperandTracker()
107 std::pair<uint32_t, uint32_t>(operandIndex, operationIndex)); in OperandTracker()
111 cb(operationIndex); in OperandTracker()
113 mUnknownInputCount[operationIndex] = count; in OperandTracker()
117 void OperandTracker::markProcessed(uint32_t operationIndex, OperationReadyCallback cb) { in markProcessed() argument
119 const Operation& operation = mModel->getOperations()[operationIndex]; in markProcessed()
229 int ExecutionStep::addOperation(int operationIndex, const ModelBuilder& fromModel) { in addOperation() argument
230 const Operation& operation = fromModel.getOperation(operationIndex); in addOperation()
821 auto enqueueOnAppropriateDevice = [&](uint32_t operationIndex) { in partitionTheWork() argument
822 int deviceIndex = bestDeviceForOperation[operationIndex]; in partitionTheWork()
823 perDeviceQueue[deviceIndex].push(operationIndex); in partitionTheWork()
824 VLOG(COMPILATION) << "enqueueOnAppropriateDevice " << operationIndex << " onto " in partitionTheWork()
860 uint32_t operationIndex = queue.front(); in partitionTheWork() local
862 int n = step->addOperation(operationIndex, *this); in partitionTheWork()
864 LOG(ERROR) << "failed to add operation " << operationIndex << " to step"; in partitionTheWork()
867 tracker.markProcessed(operationIndex, enqueueOnAppropriateDevice); in partitionTheWork()
883 uint32_t operationIndex) const { in getPerformanceInfo()
884 const Operation& operation = getOperation(operationIndex); in getPerformanceInfo()
923 bool check(size_t operationIndex) const { return mSupportsOperationByIndex[operationIndex]; } in check()
946 for (size_t operationIndex = 0; operationIndex < operationCount; operationIndex++) { in findBestDeviceForEachOperation() local
952 if (canDo[deviceIndex].check(operationIndex)) { in findBestDeviceForEachOperation()
953 const PerformanceInfo perf = getPerformanceInfo(device, operationIndex); in findBestDeviceForEachOperation()
969 << toString(getOperation(operationIndex).type); in findBestDeviceForEachOperation()
973 if (mOperations[operationIndex].type == OperationType::OEM_OPERATION) { in findBestDeviceForEachOperation()
987 (*bestDeviceForOperation)[operationIndex] = bestChoice; in findBestDeviceForEachOperation()
989 << toString(getOperation(operationIndex).type) in findBestDeviceForEachOperation()
991 << (*bestDeviceForOperation)[operationIndex]; in findBestDeviceForEachOperation()