Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc67 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
161 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
238 int recurrent_to_output_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
322 SetWeights(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
356 std::vector<float> recurrent_to_output_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
436 recurrent_to_output_weights_ = { in SetUp()
500 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
558 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
617 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
652 recurrent_to_output_weights_ = { in SetUp()
[all …]
Dunidirectional_sequence_lstm_test.cc66 recurrent_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
151 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
213 int recurrent_to_output_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
292 SetWeights(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
329 std::vector<float> recurrent_to_output_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
437 recurrent_to_output_weights_ = { in SetUp()
506 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
568 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
635 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
698 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc61 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
142 PopulateTensor(recurrent_to_output_weights_, f); in SetRecurrentToOutputWeights()
206 int recurrent_to_output_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2289 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
2370 SetData(recurrent_to_output_weights_, weight_type_, f); in SetRecurrentToOutputWeights()
2431 int recurrent_to_output_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2474 std::vector<float> recurrent_to_output_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
2554 recurrent_to_output_weights_ = { in SetUp()
2619 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
2653 recurrent_to_output_weights_ = { in SetUp()
2720 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()
3098 recurrent_to_output_weights_ = { in SetUp()
3378 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights_); in TEST_F()