Searched refs:mergeOutputs (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 162 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped() local
168 if (!mergeOutputs) { in executeTyped()
187 if (!mergeOutputs) { in executeTyped()
203 if (!mergeOutputs) { in executeTyped()
214 if (!mergeOutputs) { in executeTyped()
248 const int delta = mergeOutputs ? 1 : 0; in executeTyped()
264 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped()
286 if (mergeOutputs) { in executeTyped()
309 if (!mergeOutputs) { in executeTyped()
349 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare() local
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2871 ANeuralNetworksOperandType mergeOutputs = boolScalar; in lstmBidirectionalSequence() local
2938 mergeOutputs, in lstmBidirectionalSequence()
3983 ANeuralNetworksOperandType mergeOutputs = boolScalar; in bidirectionlSequenceRNNTest() local
3988 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3541 * * 14:mergeOutputs
3549 * input 14 (mergeOutputs). If timeMajor is set to true, then the first
3551 * [batchSize, maxTime]. If mergeOutputs is set to true, then the third
3555 * A 3-D tensor. If the input 14 (mergeOutputs) is set to true, then
|