Home
last modified time | relevance | path

Searched refs:kNumOutputs (Results 1 – 25 of 45) sorted by relevance

12

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DElementwise.cpp32 constexpr uint32_t kNumOutputs = 1; variable
87 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
99 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateAbs()
111 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateFloor()
H A DLogicalNot.cpp29 constexpr uint32_t kNumOutputs = 1; variable
46 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DRank.cpp30 constexpr uint32_t kNumOutputs = 1; variable
35 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DNeg.cpp34 constexpr uint32_t kNumOutputs = 1; variable
52 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DReduce.cpp40 constexpr uint32_t kNumOutputs = 1; variable
75 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateProdSum()
92 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateMaxMin()
115 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateLogical()
H A DElu.cpp36 constexpr uint32_t kNumOutputs = 1; variable
57 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DFill.cpp30 constexpr uint32_t kNumOutputs = 1; variable
66 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DLogicalAndOr.cpp34 constexpr uint32_t kNumOutputs = 1; variable
65 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DQuantize.cpp34 constexpr uint32_t kNumOutputs = 1; variable
68 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DSelect.cpp32 constexpr uint32_t kNumOutputs = 1; variable
71 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DSqueeze.cpp35 constexpr uint32_t kNumOutputs = 1; variable
40 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DLogSoftmax.cpp38 constexpr uint32_t kNumOutputs = 1; variable
75 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DChannelShuffle.cpp34 constexpr uint32_t kNumOutputs = 1; variable
62 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DTopK_V2.cpp34 constexpr uint32_t kNumOutputs = 2; variable
78 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DDequantize.cpp30 constexpr uint32_t kNumOutputs = 1; variable
80 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DGather.cpp34 constexpr uint32_t kNumOutputs = 1; variable
64 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DInstanceNormalization.cpp42 constexpr uint32_t kNumOutputs = 1; variable
109 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DPRelu.cpp41 constexpr uint32_t kNumOutputs = 1; variable
104 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DTranspose.cpp41 constexpr uint32_t kNumOutputs = 1; variable
79 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DLocalResponseNormalization.cpp45 constexpr uint32_t kNumOutputs = 1; variable
142 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DSlice.cpp39 constexpr uint32_t kNumOutputs = 1; variable
88 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DStridedSlice.cpp46 constexpr uint32_t kNumOutputs = 1; variable
106 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
H A DUnidirectionalSequenceRNN.cpp40 constexpr uint32_t kNumOutputs = 1; variable
134 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
H A DConcatenation.cpp43 constexpr uint32_t kNumOutputs = 1; variable
147 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
/aosp12/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
H A DFibonacciDriver.cpp47 constexpr uint32_t kNumOutputs = 1; variable
68 NN_RET_CHECK_EQ(operation.outputs.size(), kNumOutputs); in validate()

12