Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp55 constexpr uint32_t kNumOutputsMergedWithState = 3; variable
242 context->getNumOutputs() == kNumOutputsMergedWithState); in executeTyped()
325 numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in validate()
342 if (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState) { in validate()
352 NN_RET_CHECK(numOutputs == kNumOutputsMerged || numOutputs == kNumOutputsMergedWithState); in prepare()
455 (numOutputs == kNumOutputsWithState || numOutputs == kNumOutputsMergedWithState); in prepare()
/frameworks/ml/nn/common/
DUtils.cpp1113 const uint32_t kNumOutputsMergedWithState = 5; in validateOperation() local
1117 outputCount != kNumOutputsMergedWithState)) { in validateOperation()
1151 if (outputCount == kNumOutputsWithState || outputCount == kNumOutputsMergedWithState) { in validateOperation()