Searched refs:auxInput (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 110 const T* auxInput = nullptr; in executeTyped() local
116 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
120 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
163 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
169 auxInput = auxInputTransposed.data(); in executeTyped()
206 bwInput = auxInput; in executeTyped()
207 auxInput = nullptr; in executeTyped()
231 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
250 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
318 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
[all …]
|
| D | BidirectionalSequenceLSTM.cpp | 460 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval() local
464 auxInput = nullptr; in Eval()
498 GetBuffer<const float>(fw_cell_to_output_weights_), auxInput, in Eval()
550 GetBuffer<const float>(bw_cell_to_output_weights_), auxInput, in Eval()
583 const _Float16* auxInput = GetOptionalBuffer<const _Float16>(aux_input_); in Eval() local
587 auxInput = nullptr; in Eval()
622 GetOptionalBuffer<const _Float16>(fw_cell_to_output_weights_), auxInput, in Eval()
675 GetOptionalBuffer<const _Float16>(bw_cell_to_output_weights_), auxInput, in Eval()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 2897 ANeuralNetworksOperandType auxInput = floatTensor3D; in lstmBidirectionalSequence() local
2964 auxInput, in lstmBidirectionalSequence()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3678 * * 9: auxInput.
3691 * * 9: auxInput.
|