Home
last modified time | relevance | path

Searched refs:kNumOutputsWithState (Results 1 – 11 of 11) sorted by relevance

/packages/modules/NeuralNetworks/common/types/operations/src/
DBidirectionalSequenceLSTM.cpp29 const uint32_t kNumOutputsWithState = 6; in validate() local
34 context->getNumOutputs() == kNumOutputsWithState || in validate()
62 if (context->getNumOutputs() == kNumOutputsWithState || in validate()
DUnidirectionalSequenceRNN.cpp29 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
39 if (numOutputs == kNumOutputsWithState) { in validate()
DBidirectionalSequenceRNN.cpp32 numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in validate()
48 if (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState) { in validate()
DUnidirectionalSequenceLSTM.cpp29 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
71 if (context->getNumOutputs() == kNumOutputsWithState) { in validate()
/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp108 if (context->getNumOutputs() == kNumOutputsWithState) { in executeTyped()
153 if (context->getNumOutputs() == kNumOutputsWithState) { in prepare()
DBidirectionalSequenceRNN.cpp210 const bool outputState = (context->getNumOutputs() == kNumOutputsWithState || in executeTyped()
294 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in prepare()
395 (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in prepare()
DUnidirectionalSequenceLSTM.cpp271 if (context->getNumOutputs() == kNumOutputsWithState) { in prepare()
296 const bool useStateOutTensors = (context->getNumOutputs() == kNumOutputsWithState); in execute()
/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceRNN.h34 constexpr uint32_t kNumOutputsWithState = 2; variable
DBidirectionalSequenceRNN.h47 constexpr uint32_t kNumOutputsWithState = 4; variable
DUnidirectionalSequenceLSTM.h76 constexpr uint32_t kNumOutputsWithState = 3; variable
/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp948 const uint32_t kNumOutputsWithState = 6; in validateOperation() local
952 outputCount != kNumOutputsWithState && in validateOperation()
986 if (outputCount == kNumOutputsWithState || outputCount == kNumOutputsMergedWithState) { in validateOperation()