Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceRNN.cpp41 constexpr uint32_t kNumOutputsWithState = 2; variable
121 if (context->getNumOutputs() == kNumOutputsWithState) { in executeTyped()
134 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
145 if (numOutputs == kNumOutputsWithState) { in validate()
188 if (context->getNumOutputs() == kNumOutputsWithState) { in prepare()
DBidirectionalSequenceRNN.cpp54 constexpr uint32_t kNumOutputsWithState = 4; variable
241 const bool outputState = (context->getNumOutputs() == kNumOutputsWithState || in executeTyped()
325 numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in validate()
342 if (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState) { in validate()
354 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in prepare()
455 (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in prepare()
DUnidirectionalSequenceLSTM.cpp83 constexpr uint32_t kNumOutputsWithState = 3; variable
121 NN_RET_CHECK(numOutputs == kNumOutputs || numOutputs == kNumOutputsWithState); in validate()
163 if (context->getNumOutputs() == kNumOutputsWithState) { in validate()
377 if (context->getNumOutputs() == kNumOutputsWithState) { in prepare()
402 const bool useStateOutTensors = (context->getNumOutputs() == kNumOutputsWithState); in execute()
/frameworks/ml/nn/common/
DUtils.cpp1112 const uint32_t kNumOutputsWithState = 6; in validateOperation() local
1116 outputCount != kNumOutputsWithState && in validateOperation()
1151 if (outputCount == kNumOutputsWithState || outputCount == kNumOutputsMergedWithState) { in validateOperation()