Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/tools/test_generator/test_harness/
DTestHarness.cpp149 const auto& outputOperand = model.main.operands[model.main.outputIndexes[0]]; in expectMultinomialDistributionWithinTolerance() local
151 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()
/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp688 const Operand& outputOperand = *helper.getSubgraphOutputOperand(bodyModelOperand, i); in validateWhileOperation() local
689 NN_RET_CHECK(compatible(inputOperand, outputOperand)); in validateWhileOperation()
690 NN_RET_CHECK(validateControlFlowOperandUnknownSize(helper, outputOperand)); in validateWhileOperation()
1362 auto outputOperand = operands[outputIndexes[0]]; in validateOperation() local
1364 auto outputType = outputOperand.type; in validateOperation()
1398 if (inputOperand.dimensions.size() != 0 && outputOperand.dimensions.size() != 0 && in validateOperation()
1399 getNumberOfElements(outputOperand.dimensions) != 0 && in validateOperation()
1400 inputOperand.dimensions != outputOperand.dimensions) { in validateOperation()
/packages/modules/NeuralNetworks/runtime/
DExecutionPlan.cpp1825 for (const SourceOperandIndex& outputOperand : step->bodyOutputOperands) { in nextCompound() local
1827 CHECK_EQ(controller->mSourceOperandToInputIndex.count(outputOperand), 0u); in nextCompound()
1828 CHECK_EQ(controller->mSourceOperandToOutputIndex.count(outputOperand), 0u); in nextCompound()
1829 CHECK_EQ(controller->mSourceOperandToLocationOfTemporary.count(outputOperand), 1u); in nextCompound()
1830 CHECK_EQ(controller->mSourceOperandToLocationOfTemporary2.count(outputOperand), 1u); in nextCompound()
1832 std::swap(controller->mSourceOperandToLocationOfTemporary[outputOperand], in nextCompound()
1833 controller->mSourceOperandToLocationOfTemporary2[outputOperand]); in nextCompound()
DExecutionBuilder.cpp78 const Operand& outputOperand = model->getOutputOperand(outputIndex); in validateOutputShapesFromDriver() local
79 NN_RET_CHECK(!tm->isTensorType(outputOperand.type) || in validateOutputShapesFromDriver()
/packages/modules/NeuralNetworks/common/types/src/
DValidation.cpp1720 const Operand& outputOperand = getOutputOperand(subgraphs, bodyModelOperand, i); in validateWhileOperation() local
1721 NN_TRY(compatible(inputOperand, outputOperand)); in validateWhileOperation()
1723 NN_TRY(validateControlFlowOperandUnknownSize(outputOperand))); in validateWhileOperation()