Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp111 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped() local
117 if (!mergeOutputs) { in executeTyped()
136 if (!mergeOutputs) { in executeTyped()
152 if (!mergeOutputs) { in executeTyped()
163 if (!mergeOutputs) { in executeTyped()
193 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped()
216 if (mergeOutputs) { in executeTyped()
239 if (!mergeOutputs) { in executeTyped()
273 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare() local
274 if (mergeOutputs) { in prepare()
[all …]
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp1949 ANeuralNetworksOperandType mergeOutputs = boolScalar; in lstmBidirectionalSequence() local
2016 mergeOutputs, in lstmBidirectionalSequence()
2984 ANeuralNetworksOperandType mergeOutputs = boolScalar; in bidirectionlSequenceRNNTest() local
2989 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()