Searched refs:RNNStep (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 84 RNNStep<_Float16>(reinterpret_cast<_Float16*>(input_->buffer), input_->shape(), in Eval()
96 RNNStep<float>(reinterpret_cast<float*>(input_->buffer), input_->shape(), in Eval()
116 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep() function in android::nn::RNN
124 return RNNStep<T>(inputData, inputShape, /*auxInputData=*/nullptr, /*auxInputShape=*/dummyShape, in RNNStep()
136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() function in android::nn::RNN
223 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
229 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
239 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
245 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
|
| D | BidirectionalSequenceRNN.cpp | 236 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
264 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
|
| D | UnidirectionalSequenceRNN.cpp | 96 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 56 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData,
62 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData,
|