Searched refs:auxInputData (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() argument
153 bool hasAuxInput = (auxInputData != nullptr); in RNNStep()
166 aux_input_ptr_batch = auxInputData + b * aux_input_size; in RNNStep()
230 const _Float16* auxInputData, const Shape& auxInputShape,
246 const float* auxInputData, const Shape& auxInputShape,
|
| 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()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 62 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
|