Searched refs:kFirstInput (Results 1 – 6 of 6) sorted by relevance
| /packages/modules/NeuralNetworks/common/types/include/ |
| D | ControlFlow.h | 32 constexpr uint32_t kFirstInput = 3; variable
40 constexpr uint32_t kFirstInput = 2; variable
|
| /packages/modules/NeuralNetworks/common/ |
| D | LegacyUtils.cpp | 597 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + branchModelInputCount); in validateIfOperation()
601 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateIfOperation()
649 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + condModelInputCount); in validateWhileOperation()
653 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
667 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + bodyModelInputCount); in validateWhileOperation()
675 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
|
| D | CpuExecutor.cpp | 1712 operands[operation.inputs[op::kFirstInput + i]]); in executeIfOperation()
1784 operands[operation.inputs[op::kFirstInput + i]]); in executeWhileOperation()
|
| /packages/modules/NeuralNetworks/common/types/src/ |
| D | Validation.cpp | 1610 NN_RET_CHECK_EQ(inputs.size(), op::kFirstInput + branchModelInputCount); in validateIfOperation()
1614 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateIfOperation()
1673 NN_RET_CHECK_EQ(inputs.size(), op::kFirstInput + condModelInputCount); in validateWhileOperation()
1677 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
1696 NN_RET_CHECK_EQ(inputs.size(), op::kFirstInput + bodyModelInputCount); in validateWhileOperation()
1704 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
|
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionPlan.cpp | 2303 for (uint32_t i = op::kFirstInput, n = operation.inputs.size(); i < n; ++i) { in partitionTheWorkInternal()
2364 for (uint32_t i = op::kFirstInput, n = operation.inputs.size(); i < n; ++i) { in partitionTheWorkInternal()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestPartitioning.cpp | 438 CHECK_GE(operation.inputs.size(), op::kFirstInput); in getSupportedOperationsForSubgraph()
444 CHECK_GE(operation.inputs.size(), op::kFirstInput); in getSupportedOperationsForSubgraph()
|