Home
last modified time | relevance | path

Searched refs:recurrent_weights (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc66 const TfLiteTensor* recurrent_weights = in Prepare() local
79 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[0], in Prepare()
81 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[1], in Prepare()
84 TF_LITE_ENSURE_EQ(context, input_weights->type, recurrent_weights->type); in Prepare()
148 const TfLiteTensor* recurrent_weights, in EvalFloat() argument
164 const float* recurrent_weights_ptr = recurrent_weights->data.f; in EvalFloat()
176 const TfLiteTensor* recurrent_weights, in EvalHybrid() argument
198 GetInt8DataPtr(recurrent_weights, is_uint8_hybrid); in EvalHybrid()
202 float recurrent_weights_scale = recurrent_weights->params.scale; in EvalHybrid()
223 const TfLiteTensor* recurrent_weights = in Eval() local
[all …]
Dbidirectional_sequence_rnn_test.cc636 const std::initializer_list<float> recurrent_weights = { variable
795 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
796 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
834 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
835 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
871 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
872 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
910 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
911 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
955 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
[all …]
Dunidirectional_sequence_rnn.cc61 const TfLiteTensor* recurrent_weights = in Prepare() local
79 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[0], in Prepare()
81 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[1], in Prepare()
84 TF_LITE_ENSURE_EQ(context, input_weights->type, recurrent_weights->type); in Prepare()
148 const TfLiteTensor* recurrent_weights, in EvalFloat() argument
165 const float* recurrent_weights_ptr = recurrent_weights->data.f; in EvalFloat()
206 const TfLiteTensor* recurrent_weights, const TfLiteTensor* bias, in EvalHybrid() argument
231 reinterpret_cast<const int8_t*>(recurrent_weights->data.uint8); in EvalHybrid()
237 recurrent_weights_ptr = recurrent_weights->data.int8; in EvalHybrid()
244 float recurrent_weights_scale = recurrent_weights->params.scale; in EvalHybrid()
[all …]
Dbasic_rnn_test.cc178 const TensorType& recurrent_weights = TensorType_FLOAT32) in RNNOpModel() argument
182 recurrent_weights_ = AddInput(recurrent_weights); in RNNOpModel()
Dunidirectional_sequence_rnn_test.cc177 const TensorType& recurrent_weights = TensorType_FLOAT32) in UnidirectionalRNNOpModel() argument
184 recurrent_weights_ = AddInput(recurrent_weights); in UnidirectionalRNNOpModel()
/external/libopus/src/
Dmlp.c115 sum += gru->recurrent_weights[j*stride + i]*state[j]; in compute_gru()
125 sum += gru->recurrent_weights[N + j*stride + i]*state[j]; in compute_gru()
135 sum += gru->recurrent_weights[2*N + j*stride + i]*state[j]*r[j]; in compute_gru()
Dmlp.h47 const opus_int8 *recurrent_weights; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1905 const TensorType recurrent_weights = TensorType_FLOAT32) in RNNOpModel() argument
1909 recurrent_weights_ = AddInput(recurrent_weights); in RNNOpModel()