Searched refs:ioIndex (Results 1 – 9 of 9) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | ValidateHal.cpp | 890 NN_RET_CHECK_LT(role.ioIndex, inputIndexes.size()); in validateMemoryDesc()
893 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::INPUT, role.ioIndex); in validateMemoryDesc()
895 operands.push_back(model->main.operands[inputIndexes[role.ioIndex]]); in validateMemoryDesc()
904 NN_RET_CHECK_LT(role.ioIndex, outputIndexes.size()); in validateMemoryDesc()
907 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::OUTPUT, role.ioIndex); in validateMemoryDesc()
909 operands.push_back(model->main.operands[outputIndexes[role.ioIndex]]); in validateMemoryDesc()
|
| /packages/modules/NeuralNetworks/shim_and_sl/ |
| D | ShimDevice.cpp | 374 slDesc, pmodel->getCompilation().getHandle(), role.ioIndex, role.probability); in allocate()
383 const auto& op = model.getOperands()[model.getInputs()[role.ioIndex]]; in allocate()
412 slDesc, pmodel->getCompilation().getHandle(), role.ioIndex, role.probability); in allocate()
420 const auto& op = model.getOperands()[model.getOutputs()[role.ioIndex]]; in allocate()
|
| /packages/modules/NeuralNetworks/runtime/test/fuzzing/ |
| D | RandomGraphGenerator.h | 67 int32_t ioIndex = -1; member
|
| /packages/modules/NeuralNetworks/common/types/src/ |
| D | Validation.cpp | 1264 NN_RET_CHECK_LT(role.ioIndex, inputIndexes.size()); in validateMemoryDescImpl()
1267 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::INPUT, role.ioIndex); in validateMemoryDescImpl()
1269 operands.push_back(model->main.operands[inputIndexes[role.ioIndex]]); in validateMemoryDescImpl()
1278 NN_RET_CHECK_LT(role.ioIndex, outputIndexes.size()); in validateMemoryDescImpl()
1281 const auto [it, success] = roles.emplace(preparedModel.get(), IOType::OUTPUT, role.ioIndex); in validateMemoryDescImpl()
1283 operands.push_back(model->main.operands[outputIndexes[role.ioIndex]]); in validateMemoryDescImpl()
|
| D | TypeUtils.cpp | 687 << ", .ioIndex=" << bufferRole.ioIndex << ", .probability=" << bufferRole.probability in operator <<()
|
| /packages/modules/NeuralNetworks/common/types/include/nnapi/ |
| D | Types.h | 852 uint32_t ioIndex = 0; member
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | Memory.cpp | 386 BufferRole role = {.modelIndex = modelIndex, .ioIndex = ind, .probability = prob}; in addRole()
|
| D | ExecutionPlan.cpp | 2083 for (const auto& [stepIndex, type, ioIndex] : it->second) { in forEachStepRoleOfSourceOperand()
2086 callback(step->getPreparedStepModel().get(), type, ioIndex); in forEachStepRoleOfSourceOperand()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 7916 uint32_t ioIndex%{init_int};
|