Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc68 const TfLiteTensor* recurrent_weights; in Prepare() local
71 GetInputSafe(context, node, kRecurrentWeightsTensor, &recurrent_weights)); in Prepare()
85 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[0], in Prepare()
87 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[1], in Prepare()
91 recurrent_weights->type); in Prepare()
202 const TfLiteTensor* recurrent_weights, in EvalFloat() argument
218 const float* recurrent_weights_ptr = GetTensorData<float>(recurrent_weights); in EvalFloat()
230 const TfLiteTensor* recurrent_weights, in EvalHybrid() argument
252 GetTensorData<int8_t>(recurrent_weights); in EvalHybrid()
256 float recurrent_weights_scale = recurrent_weights->params.scale; in EvalHybrid()
[all …]
Dbidirectional_sequence_rnn_test.cc645 const std::initializer_list<float> recurrent_weights = { variable
874 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST_P()
875 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST_P()
922 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST_P()
923 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST_P()
969 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST_P()
970 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST_P()
1010 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
1011 rnn.SetBwRecurrentWeights(recurrent_weights); in TEST()
1056 rnn.SetFwRecurrentWeights(recurrent_weights); in TEST()
[all …]
Dunidirectional_sequence_rnn.cc69 const TfLiteTensor* recurrent_weights; in Prepare() local
72 GetInputSafe(context, node, kRecurrentWeightsTensor, &recurrent_weights)); in Prepare()
91 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[0], in Prepare()
93 TF_LITE_ENSURE_EQ(context, recurrent_weights->dims->data[1], in Prepare()
97 recurrent_weights->type); in Prepare()
209 const TfLiteTensor* recurrent_weights, in EvalFloat() argument
226 const float* recurrent_weights_ptr = GetTensorData<float>(recurrent_weights); in EvalFloat()
270 const TfLiteTensor* recurrent_weights, const TfLiteTensor* bias, in EvalHybrid() argument
291 GetTensorData<int8_t>(recurrent_weights); in EvalHybrid()
298 float recurrent_weights_scale = recurrent_weights->params.scale; in EvalHybrid()
[all …]
Dbasic_rnn_test.cc175 const TensorType& recurrent_weights = TensorType_FLOAT32, in RNNOpModel() argument
180 recurrent_weights_ = AddInput(recurrent_weights); in RNNOpModel()
Dunidirectional_sequence_rnn_test.cc176 const TensorType& recurrent_weights = TensorType_FLOAT32, in UnidirectionalRNNOpModel() argument
184 recurrent_weights_ = AddInput(recurrent_weights); in UnidirectionalRNNOpModel()
/external/webrtc/modules/audio_processing/agc2/rnn_vad/
Drnn.cc128 rtc::ArrayView<const float> recurrent_weights, in ComputeGruUpdateResetGates() argument
139 gate[o] += state[s] * recurrent_weights[o * output_size + s]; in ComputeGruUpdateResetGates()
148 rtc::ArrayView<const float> recurrent_weights, in ComputeGruOutputGate() argument
160 gate[o] += state[s] * recurrent_weights[o * output_size + s] * reset[s]; in ComputeGruOutputGate()
171 rtc::ArrayView<const float> recurrent_weights, in ComputeGruLayerOutput() argument
183 recurrent_weights.subview(0, stride_out), bias.subview(0, output_size), in ComputeGruLayerOutput()
190 recurrent_weights.subview(stride_out, stride_out), in ComputeGruLayerOutput()
197 recurrent_weights.subview(2 * stride_out, stride_out), in ComputeGruLayerOutput()
323 const rtc::ArrayView<const int8_t> recurrent_weights, in GatedRecurrentLayer() argument
330 GetPreprocessedGruTensor(recurrent_weights, output_size)), in GatedRecurrentLayer()
Drnn.h79 rtc::ArrayView<const int8_t> recurrent_weights,
/external/libopus/src/
Dmlp.c120 gemm_accum(z, gru->recurrent_weights, N, N, stride, state); in compute_gru()
128 gemm_accum(r, &gru->recurrent_weights[N], N, N, stride, state); in compute_gru()
138 gemm_accum(h, &gru->recurrent_weights[2*N], N, N, stride, tmp); in compute_gru()
Dmlp.h47 const opus_int8 *recurrent_weights; member
/external/rnnoise/src/
Drnn.c127 sum += gru->recurrent_weights[j*stride + i]*state[j]; in compute_gru()
137 sum += gru->recurrent_weights[N + j*stride + i]*state[j]; in compute_gru()
147 sum += gru->recurrent_weights[2*N + j*stride + i]*state[j]*r[j]; in compute_gru()
Drnn_reader.c126 INPUT_ARRAY(name->recurrent_weights, name->nb_neurons * name->nb_neurons * 3); \ in rnnoise_model_from_file()
153 free((void *) model->name->recurrent_weights); \ in rnnoise_model_free()
Drnn.h55 const rnn_weight *recurrent_weights; member
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2737 const TensorType recurrent_weights = TensorType_FLOAT32) in RNNOpModel() argument
2741 recurrent_weights_ = AddInput(recurrent_weights); in RNNOpModel()