Searched refs:mergeOutputs (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 131 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped() local
137 if (!mergeOutputs) { in executeTyped()
156 if (!mergeOutputs) { in executeTyped()
172 if (!mergeOutputs) { in executeTyped()
183 if (!mergeOutputs) { in executeTyped()
217 const int delta = mergeOutputs ? 1 : 0; in executeTyped()
233 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped()
255 if (mergeOutputs) { in executeTyped()
278 if (!mergeOutputs) { in executeTyped()
289 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare() local
[all …]
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2909 ANeuralNetworksOperandType mergeOutputs = boolScalar; in lstmBidirectionalSequence() local
2976 mergeOutputs, in lstmBidirectionalSequence()
4021 ANeuralNetworksOperandType mergeOutputs = boolScalar; in bidirectionlSequenceRNNTest() local
4026 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3704 * * 14:mergeOutputs
3712 * input 14 (mergeOutputs). If timeMajor is set to true, then the first
3714 * [batchSize, maxTime]. If mergeOutputs is set to true, then the third
3718 * A 3-D tensor. If the input 14 (mergeOutputs) is set to true, then
|