Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/operations/src/
DBidirectionalSequenceLSTM.cpp30 const uint32_t kNumOutputsMergedWithState = 5; in validate() local
35 context->getNumOutputs() == kNumOutputsMergedWithState)) in validate()
63 context->getNumOutputs() == kNumOutputsMergedWithState) { in validate()
DBidirectionalSequenceRNN.cpp32 numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in validate()
48 if (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState) { in validate()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceRNN.h48 constexpr uint32_t kNumOutputsMergedWithState = 3; variable
/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp211 context->getNumOutputs() == kNumOutputsMergedWithState); in executeTyped()
292 NN_RET_CHECK(numOutputs == kNumOutputsMerged || numOutputs == kNumOutputsMergedWithState); in prepare()
395 (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in prepare()
/packages/modules/NeuralNetworks/common/
DLegacyUtils.cpp949 const uint32_t kNumOutputsMergedWithState = 5; in validateOperation() local
953 outputCount != kNumOutputsMergedWithState)) { in validateOperation()
986 if (outputCount == kNumOutputsWithState || outputCount == kNumOutputsMergedWithState) { in validateOperation()