Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DElementwise.cpp33 constexpr uint32_t kNumOutputs = 1; variable
90 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
102 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateAbs()
115 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateFloor()
DRank.cpp31 constexpr uint32_t kNumOutputs = 1; variable
36 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLogicalNot.cpp30 constexpr uint32_t kNumOutputs = 1; variable
49 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DNeg.cpp35 constexpr uint32_t kNumOutputs = 1; variable
55 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DElu.cpp39 constexpr uint32_t kNumOutputs = 1; variable
60 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DFill.cpp31 constexpr uint32_t kNumOutputs = 1; variable
69 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DReduce.cpp39 constexpr uint32_t kNumOutputs = 1; variable
74 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateProdSum()
91 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateMaxMin()
114 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validateLogical()
DLogicalAndOr.cpp32 constexpr uint32_t kNumOutputs = 1; variable
65 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DQuantize.cpp35 constexpr uint32_t kNumOutputs = 1; variable
71 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSqueeze.cpp36 constexpr uint32_t kNumOutputs = 1; variable
43 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLogSoftmax.cpp39 constexpr uint32_t kNumOutputs = 1; variable
76 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSelect.cpp33 constexpr uint32_t kNumOutputs = 1; variable
74 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DTopK_V2.cpp35 constexpr uint32_t kNumOutputs = 2; variable
81 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DChannelShuffle.cpp37 constexpr uint32_t kNumOutputs = 1; variable
65 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DGather.cpp35 constexpr uint32_t kNumOutputs = 1; variable
67 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DTranspose.cpp40 constexpr uint32_t kNumOutputs = 1; variable
78 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DDequantize.cpp31 constexpr uint32_t kNumOutputs = 1; variable
81 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DInstanceNormalization.cpp40 constexpr uint32_t kNumOutputs = 1; variable
107 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DPRelu.cpp42 constexpr uint32_t kNumOutputs = 1; variable
103 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DSlice.cpp37 constexpr uint32_t kNumOutputs = 1; variable
86 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DStridedSlice.cpp44 constexpr uint32_t kNumOutputs = 1; variable
104 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DLocalResponseNormalization.cpp43 constexpr uint32_t kNumOutputs = 1; variable
139 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
DUnidirectionalSequenceRNN.cpp40 constexpr uint32_t kNumOutputs = 1; variable
134 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
DConcatenation.cpp40 constexpr uint32_t kNumOutputs = 1; variable
144 NN_RET_CHECK_EQ(context->getNumOutputs(), kNumOutputs); in validate()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp48 constexpr uint32_t kNumOutputs = 1; variable
69 NN_RET_CHECK_EQ(operation.outputs.size(), kNumOutputs); in validate()

12