Home
last modified time | relevance | path

Searched refs:kFirstInput (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/include/
DControlFlow.h27 constexpr uint32_t kFirstInput = 3; variable
35 constexpr uint32_t kFirstInput = 2; variable
/frameworks/ml/nn/common/
DUtils.cpp742 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + branchModelInputCount); in validateIfOperation()
746 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateIfOperation()
795 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + condModelInputCount); in validateWhileOperation()
799 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
813 NN_RET_CHECK_EQ(inputCount, op::kFirstInput + bodyModelInputCount); in validateWhileOperation()
821 const Operand& outerOperand = operands[inputs[op::kFirstInput + i]]; in validateWhileOperation()
DCpuExecutor.cpp1761 operands[operation.inputs[op::kFirstInput + i]]); in executeIfOperation()
1832 operands[operation.inputs[op::kFirstInput + i]]); in executeWhileOperation()
/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1755 for (uint32_t i = op::kFirstInput, n = operation.inputs.size(); i < n; ++i) { in partitionTheWorkInternal()
1816 for (uint32_t i = op::kFirstInput, n = operation.inputs.size(); i < n; ++i) { in partitionTheWorkInternal()
/frameworks/ml/nn/runtime/test/
DTestPartitioning.cpp456 CHECK_GE(operation.inputs.size(), op::kFirstInput); in getSupportedOperationsForSubgraph()
462 CHECK_GE(operation.inputs.size(), op::kFirstInput); in getSupportedOperationsForSubgraph()