Searched refs:auxInputData (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNN.cpp | 140 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() argument
157 bool hasAuxInput = (auxInputData != nullptr); in RNNStep()
170 aux_input_ptr_batch = auxInputData + b * aux_input_size; in RNNStep()
234 const _Float16* auxInputData, const Shape& auxInputShape,
250 const float* auxInputData, const Shape& auxInputShape,
|
| D | RNN.h | 56 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
|
| D | LSTM.cpp | 465 const float* auxInputData = in LSTMEvalFloat32() local
476 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()
701 const float* auxInputData = in LSTMEvalFloat16() local
714 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()
|