Searched refs:auxInput (Results 1 – 6 of 6) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | BidirectionalSequenceRNN.cpp | 139 const T* auxInput = nullptr; in executeTyped() local
145 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
149 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
198 auxInput = auxInputTransposed.data(); in executeTyped()
235 bwInput = auxInput; in executeTyped()
236 auxInput = nullptr; in executeTyped()
260 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
279 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
377 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
420 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
[all …]
|
| H A 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()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/ |
| H A D | TestValidateOperations.cpp | 2858 ANeuralNetworksOperandType auxInput = floatTensor3D; in lstmBidirectionalSequence() local
2925 auxInput, in lstmBidirectionalSequence()
|
| /aosp12/packages/modules/NeuralNetworks/tools/api/ |
| H A D | types.spec | 3621 * * 9: auxInput.
3634 * * 9: auxInput.
|
| /aosp12/hardware/interfaces/neuralnetworks/1.2/ |
| H A D | types.hal | 2635 * * 9: auxInput.
|
| /aosp12/hardware/interfaces/neuralnetworks/1.3/ |
| H A D | types.hal | 2778 * * 9: auxInput.
|