Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DValidateHal.cpp452 auto isValidSubgraphReference = [&subgraphs](const Operand& modelOperand) -> bool { in validateOperations() argument
453 NN_RET_CHECK(modelOperand.type == OperandType::SUBGRAPH) in validateOperations()
454 << "Unexpected operand type: " << toString(modelOperand.type); in validateOperations()
455 NN_RET_CHECK_LT(modelOperand.location.offset, subgraphs.size()) in validateOperations()
459 auto getSubgraph = [&subgraphs](const Operand& modelOperand) -> const Subgraph* { in validateOperations() argument
460 CHECK_LT(modelOperand.location.offset, subgraphs.size()); in validateOperations()
461 return &subgraphs[modelOperand.location.offset]; in validateOperations()
463 auto getInputCount = [&getSubgraph](const Operand& modelOperand) -> uint32_t { in validateOperations() argument
464 return getSubgraph(modelOperand)->inputIndexes.size(); in validateOperations()
466 auto getOutputCount = [&getSubgraph](const Operand& modelOperand) -> uint32_t { in validateOperations() argument
[all …]
/frameworks/ml/nn/runtime/
DModelBuilder.cpp372 auto isValidSubgraphReference = [this](const Operand& modelOperand) -> bool { in addOperation() argument
373 NN_RET_CHECK(modelOperand.type == OperandType::SUBGRAPH) in addOperation()
374 << "Unexpected operand type: " << toString(modelOperand.type); in addOperation()
375 NN_RET_CHECK_LT(modelOperand.location.offset, referencedModelCount()) in addOperation()
379 auto getInputCount = [this](const Operand& modelOperand) -> uint32_t { in addOperation() argument
380 return getReferencedModel(modelOperand)->inputCount(); in addOperation()
382 auto getOutputCount = [this](const Operand& modelOperand) -> uint32_t { in addOperation() argument
383 return getReferencedModel(modelOperand)->outputCount(); in addOperation()
385 auto getInputOperand = [this](const Operand& modelOperand, uint32_t index) -> const Operand* { in addOperation() argument
386 return &getReferencedModel(modelOperand)->getInputOperand(index); in addOperation()
[all …]