Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc66 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
157 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
237 int recurrent_to_cell_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
318 SetWeights(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
354 std::vector<float> recurrent_to_cell_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
424 recurrent_to_cell_weights_ = { in SetUp()
498 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
556 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
615 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
640 recurrent_to_cell_weights_ = { in SetUp()
[all …]
Dunidirectional_sequence_lstm_test.cc65 recurrent_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
147 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
212 int recurrent_to_cell_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
288 SetWeights(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
327 std::vector<float> recurrent_to_cell_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
425 recurrent_to_cell_weights_ = { in SetUp()
504 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
566 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
633 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
696 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc60 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
138 PopulateTensor(recurrent_to_cell_weights_, f); in SetRecurrentToCellWeights()
205 int recurrent_to_cell_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2288 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
2366 SetData(recurrent_to_cell_weights_, weight_type_, f); in SetRecurrentToCellWeights()
2430 int recurrent_to_cell_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2472 std::vector<float> recurrent_to_cell_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
2542 recurrent_to_cell_weights_ = { in SetUp()
2617 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
2641 recurrent_to_cell_weights_ = { in SetUp()
2718 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()
2934 recurrent_to_cell_weights_ = { in SetUp()
3376 lstm.SetRecurrentToCellWeights(recurrent_to_cell_weights_); in TEST_F()