Searched refs:RNNStep (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNN.cpp | 88 RNNStep<_Float16>(reinterpret_cast<_Float16*>(input_->buffer), input_->shape(), in Eval()
100 RNNStep<float>(reinterpret_cast<float*>(input_->buffer), input_->shape(), in Eval()
120 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep() function in android::nn::RNN
128 return RNNStep<T>(inputData, inputShape, /*auxInputData=*/nullptr, /*auxInputShape=*/dummyShape, in RNNStep()
140 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() function in android::nn::RNN
227 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
233 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
243 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
249 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
|
| D | RNN.h | 50 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData,
56 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
|
| D | BidirectionalSequenceRNN.cpp | 267 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
295 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
|
| D | UnidirectionalSequenceRNN.cpp | 109 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
|