Searched refs:auxInput (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 141 const T* auxInput = nullptr; in executeTyped() local
147 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
151 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
194 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
200 auxInput = auxInputTransposed.data(); in executeTyped()
237 bwInput = auxInput; in executeTyped()
238 auxInput = nullptr; in executeTyped()
262 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
281 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
378 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
[all …]
|
| D | BidirectionalSequenceLSTM.cpp | 461 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval() local
465 auxInput = nullptr; in Eval()
499 GetBuffer<const float>(fw_cell_to_output_weights_), auxInput, in Eval()
551 GetBuffer<const float>(bw_cell_to_output_weights_), auxInput, in Eval()
584 const _Float16* auxInput = GetOptionalBuffer<const _Float16>(aux_input_); in Eval() local
588 auxInput = nullptr; in Eval()
623 GetOptionalBuffer<const _Float16>(fw_cell_to_output_weights_), auxInput, in Eval()
676 GetOptionalBuffer<const _Float16>(bw_cell_to_output_weights_), auxInput, in Eval()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2859 ANeuralNetworksOperandType auxInput = floatTensor3D; in lstmBidirectionalSequence() local
2926 auxInput, in lstmBidirectionalSequence()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3515 * * 9: auxInput.
3528 * * 9: auxInput.
|