Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc75 cell_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
79 cell_to_forget_weights_ = AddNullInput(); in LSTMOpModel()
169 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
241 int cell_to_forget_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
330 SetWeights(cell_to_forget_weights_, f); in SetCellToForgetWeights()
358 std::vector<float> cell_to_forget_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
658 cell_to_forget_weights_ = {0.47485286, -0.51955009, -0.24458408, in SetUp()
720 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
779 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
838 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc74 cell_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
78 cell_to_forget_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel()
159 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
216 int cell_to_forget_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
300 SetWeights(cell_to_forget_weights_, f); in SetCellToForgetWeights()
331 std::vector<float> cell_to_forget_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
739 cell_to_forget_weights_ = {0.47485286, -0.51955009, -0.24458408, in SetUp()
806 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
869 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
931 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc69 cell_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
73 cell_to_forget_weights_ = AddNullInput(); in LSTMOpModel()
150 PopulateTensor(cell_to_forget_weights_, f); in SetCellToForgetWeights()
209 int cell_to_forget_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2297 cell_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
2301 cell_to_forget_weights_ = AddNullInput(); in LSTMOpModel()
2378 SetData(cell_to_forget_weights_, weight_type_, f); in SetCellToForgetWeights()
2434 int cell_to_forget_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2476 std::vector<float> cell_to_forget_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
2659 cell_to_forget_weights_ = {0.47485286, -0.51955009, -0.24458408, in SetUp()
2722 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
3187 cell_to_forget_weights_ = { in SetUp()
3381 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()