Home
last modified time | relevance | path

Searched refs:innerOperand (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1190 const SourceOperandIndex& innerOperand) { in setInput() argument
1191 VLOG(EXECUTION) << "mapping input " << toString(innerOperand) << " from " in setInput()
1194 CHECK_LE(mSourceOperandToOffsetOfTemporary.count(innerOperand) + in setInput()
1195 mSourceOperandToInputIndex.count(innerOperand) + in setInput()
1196 mSourceOperandToOutputIndex.count(innerOperand) + in setInput()
1197 mSourceOperandToConstantReference.count(innerOperand), in setInput()
1200 mSourceOperandToOffsetOfTemporary.erase(innerOperand); in setInput()
1201 mSourceOperandToInputIndex.erase(innerOperand); in setInput()
1202 mSourceOperandToOutputIndex.erase(innerOperand); in setInput()
1203 mSourceOperandToConstantReference.erase(innerOperand); in setInput()
[all …]
DExecutionPlan.h460 const SourceOperandIndex& innerOperand);
462 const SourceOperandIndex& innerOperand);
/frameworks/ml/nn/common/
DUtils.cpp745 const Operand& innerOperand = *helper.getSubgraphInputOperand(branchModelOperand, i); in validateIfOperation() local
747 NN_RET_CHECK(compatible(innerOperand, outerOperand)); in validateIfOperation()
750 const Operand& innerOperand = *helper.getSubgraphOutputOperand(branchModelOperand, i); in validateIfOperation() local
752 NN_RET_CHECK(compatible(innerOperand, outerOperand)); in validateIfOperation()
798 const Operand& innerOperand = *helper.getSubgraphInputOperand(condModelOperand, i); in validateWhileOperation() local
800 NN_RET_CHECK(compatible(innerOperand, outerOperand)); in validateWhileOperation()
801 NN_RET_CHECK(validateControlFlowOperandUnknownSize(helper, innerOperand)); in validateWhileOperation()
820 const Operand& innerOperand = *helper.getSubgraphInputOperand(bodyModelOperand, i); in validateWhileOperation() local
822 NN_RET_CHECK(compatible(innerOperand, outerOperand)); in validateWhileOperation()
823 NN_RET_CHECK(validateControlFlowOperandUnknownSize(helper, innerOperand)); in validateWhileOperation()
[all …]
DCpuExecutor.cpp1897 RunTimeOperandInfo& innerOperand = condOperands[condSubgraph.inputIndexes[i]]; in executeWhileOperation() local
1898 if (int error; !setInfoAndAllocateIfNeeded(&outerOperand, innerOperand.shape(), &error)) { in executeWhileOperation()
1901 CHECK_EQ(outerOperand.length, innerOperand.length); in executeWhileOperation()
1903 std::memcpy(outerOperand.buffer, innerOperand.buffer, innerOperand.length); in executeWhileOperation()