Home
last modified time | relevance | path

Searched refs:getNumOutputs (Results 1 – 25 of 75) sorted by relevance

123

/packages/modules/NeuralNetworks/common/types/operations/src/
DBidirectionalSequenceLSTM.cpp32 (context->getNumOutputs() == kNumOutputs || in validate()
33 context->getNumOutputs() == kNumOutputsMerged || in validate()
34 context->getNumOutputs() == kNumOutputsWithState || in validate()
35 context->getNumOutputs() == kNumOutputsMergedWithState)) in validate()
37 << ", expected 61) or output operands (" << context->getNumOutputs() in validate()
62 if (context->getNumOutputs() == kNumOutputsWithState || in validate()
63 context->getNumOutputs() == kNumOutputsMergedWithState) { in validate()
66 std::vector<OperandType> outExpectedTypes(context->getNumOutputs(), inputType); in validate()
DReshape.cpp29 context->getNumOutputs() == 1) in validateDepthToSpace()
31 << ", expected 3 or 2) or output operands (" << context->getNumOutputs() in validateDepthToSpace()
69 context->getNumOutputs() == 1) in validateSpaceToDepth()
71 << ", expected 3 or 2) or output operands (" << context->getNumOutputs() in validateSpaceToDepth()
108 NN_RET_CHECK(context->getNumInputs() == 2 && context->getNumOutputs() == 1) in validatePad()
156 NN_RET_CHECK(context->getNumInputs() == 3 && context->getNumOutputs() == 1) in validatePadV2()
204 context->getNumOutputs() == 1) in validateBatchToSpaceND()
206 << ", expected 3 or 2) or output operands (" << context->getNumOutputs() in validateBatchToSpaceND()
254 context->getNumOutputs() == 1) in validateSpaceToBatchND()
256 << ", expected 4 or 3) or output operands (" << context->getNumOutputs() in validateSpaceToBatchND()
[all …]
DElementwise.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
38 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateAbs()
50 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateFloor()
69 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateRsqrt()
DReduce.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateProdSum()
43 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateMaxMin()
66 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateLogical()
DGenerateProposals.cpp30 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
58 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
107 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
172 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLSTM.cpp29 context->getNumOutputs() == 4) in validate()
31 << ", expected 23 or 27) or output operands (" << context->getNumOutputs() in validate()
DGroupedConv2D.cpp28 context->getNumOutputs() == 1) in validate()
30 << ", expected 12 or 9) or output operands (" << context->getNumOutputs() in validate()
DActivation.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
55 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateHardSwish()
DUnidirectionalSequenceLSTM.cpp28 const uint32_t numOutputs = context->getNumOutputs(); in validate()
71 if (context->getNumOutputs() == kNumOutputsWithState) { in validate()
DLogicalNot.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLogicalAndOr.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DNeg.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DQuantize.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DElu.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DPRelu.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSelect.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DMultinomial.cpp27 NN_RET_CHECK(context->getNumInputs() == 3 && context->getNumOutputs() == 1) in validate()
DHashtableLookup.cpp27 NN_RET_CHECK(context->getNumInputs() == 3 && context->getNumOutputs() == 2) in validate()
DDensify.cpp32 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DPow.cpp27 NN_RET_CHECK(context->getNumInputs() == 2 && context->getNumOutputs() == 1) in validate()
DGather.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DTopK_V2.cpp26 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSVDF.cpp27 NN_RET_CHECK(context->getNumInputs() == 7 && context->getNumOutputs() == 2) in validate()
/packages/modules/NeuralNetworks/common/types/src/
DOperationsValidationUtils.cpp50 << ") or output operands (" << getNumOutputs() << ", expected " << expOut in invalidInOutNumberMessage()
61 NN_RET_CHECK_EQ(getNumOutputs(), outExpectedInTypes.size()) in validateOperationOperandTypes()
63 << getNumOutputs() << " outputs"; in validateOperationOperandTypes()
69 for (size_t i = 0; i < getNumOutputs(); i++) { in validateOperationOperandTypes()
87 expectedTypes, "output", context->getNumOutputs(), in validateOutputTypes()
103 for (uint32_t i = 0, n = context->getNumOutputs(); i < n; ++i) { in validateVersion()
/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp108 if (context->getNumOutputs() == kNumOutputsWithState) { in executeTyped()
153 if (context->getNumOutputs() == kNumOutputsWithState) { in prepare()

123