Searched defs:hidden_state (Results 1 – 5 of 5) sorted by relevance
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLRNNLayer.cpp | 47 …ts, const ITensorInfo *recurrent_weights, const ITensorInfo *bias, const ITensorInfo *hidden_state, in validate()
76 …ghts, const ICLTensor *recurrent_weights, const ICLTensor *bias, ICLTensor *hidden_state, ICLTenso… in configure()
83 ICLTensor *hidden_state, in configure()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NERNNLayer.cpp | 45 …ts, const ITensorInfo *recurrent_weights, const ITensorInfo *bias, const ITensorInfo *hidden_state, in validate()
72 … *weights, const ITensor *recurrent_weights, const ITensor *bias, ITensor *hidden_state, ITensor *… in configure()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | basic_rnn.cc | 74 const TfLiteTensor* hidden_state; in Prepare() local
205 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
236 TfLiteTensor* hidden_state, TfLiteTensor* output, in EvalHybrid()
295 TfLiteTensor* hidden_state = in Eval() local
|
| D | unidirectional_sequence_rnn.cc | 75 const TfLiteTensor* hidden_state; in Prepare() local
212 TfLiteTensor* hidden_state, TfLiteTensor* output) { in EvalFloat()
273 TfLiteTensor* hidden_state, TfLiteTensor* output, TfLiteTensor* zero_points, in EvalHybrid()
370 TfLiteTensor* hidden_state = in Eval() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | RNNLayerFixture.h | 72 TensorType hidden_state = create_tensor<TensorType>(output_shape, data_type); in compute_target() local
|