Searched refs:auxInputData (Results 1 – 3 of 3) sorted by relevance
140 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() argument157 bool hasAuxInput = (auxInputData != nullptr); in RNNStep()170 aux_input_ptr_batch = auxInputData + b * aux_input_size; in RNNStep()
54 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
422 const float* auxInputData = in LSTMEvalFloat32() local433 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()658 const float* auxInputData = in LSTMEvalFloat16() local671 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()