Searched refs:outputOperand (Results 1 – 3 of 3) sorted by relevance
149 const auto& outputOperand = model.main.operands[model.main.outputIndexes[0]]; in expectMultinomialDistributionWithinTolerance() local151 ASSERT_EQ(outputOperand.dimensions.size(), 2u); in expectMultinomialDistributionWithinTolerance()155 const int kNumSamples = outputOperand.dimensions[1]; in expectMultinomialDistributionWithinTolerance()157 const uint32_t outputLength = getNumberOfElements(outputOperand); in expectMultinomialDistributionWithinTolerance()
1367 for (const SourceOperandIndex& outputOperand : step->bodyOutputOperands) { in nextCompound() local1369 CHECK_EQ(controller->mSourceOperandToInputIndex.count(outputOperand), 0u); in nextCompound()1370 CHECK_EQ(controller->mSourceOperandToOutputIndex.count(outputOperand), 0u); in nextCompound()1371 CHECK_EQ(controller->mSourceOperandToOffsetOfTemporary.count(outputOperand), 1u); in nextCompound()1372 CHECK_EQ(controller->mSourceOperandToOffsetOfTemporary2.count(outputOperand), 1u); in nextCompound()1374 std::swap(controller->mSourceOperandToOffsetOfTemporary[outputOperand], in nextCompound()1375 controller->mSourceOperandToOffsetOfTemporary2[outputOperand]); in nextCompound()
834 const Operand& outputOperand = *helper.getSubgraphOutputOperand(bodyModelOperand, i); in validateWhileOperation() local835 NN_RET_CHECK(compatible(inputOperand, outputOperand)); in validateWhileOperation()836 NN_RET_CHECK(validateControlFlowOperandUnknownSize(helper, outputOperand)); in validateWhileOperation()1537 auto outputOperand = operands[outputIndexes[0]]; in validateOperation() local1539 auto outputType = outputOperand.type; in validateOperation()1573 if (inputOperand.dimensions.size() != 0 && outputOperand.dimensions.size() != 0 && in validateOperation()1574 getNumberOfElements(outputOperand.dimensions) != 0 && in validateOperation()1575 inputOperand.dimensions != outputOperand.dimensions) { in validateOperation()