Home
last modified time | relevance | path

Searched refs:recurrent_to_forget_weights_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc65 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
153 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
236 int recurrent_to_forget_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
314 SetWeights(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
355 std::vector<float> recurrent_to_forget_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
430 recurrent_to_forget_weights_ = { in SetUp()
499 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
557 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
616 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
646 recurrent_to_forget_weights_ = { in SetUp()
[all …]
Dunidirectional_sequence_lstm_test.cc64 recurrent_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
143 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
211 int recurrent_to_forget_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
284 SetWeights(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
328 std::vector<float> recurrent_to_forget_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
431 recurrent_to_forget_weights_ = { in SetUp()
505 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
567 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
634 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
697 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc59 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
134 PopulateTensor(recurrent_to_forget_weights_, f); in SetRecurrentToForgetWeights()
204 int recurrent_to_forget_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2287 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
2362 SetData(recurrent_to_forget_weights_, weight_type_, f); in SetRecurrentToForgetWeights()
2429 int recurrent_to_forget_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2473 std::vector<float> recurrent_to_forget_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
2548 recurrent_to_forget_weights_ = { in SetUp()
2618 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
2647 recurrent_to_forget_weights_ = { in SetUp()
2719 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()
3016 recurrent_to_forget_weights_ = { in SetUp()
3377 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights_); in TEST_F()