Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Doptional_tensor_test.cc65 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
67 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
72 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
146 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
208 int cell_to_input_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
Dlstm_test.cc71 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
73 cell_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
78 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
165 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
240 int cell_to_input_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
326 SetWeights(cell_to_input_weights_, f); in SetCellToInputWeights()
357 std::vector<float> cell_to_input_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
1297 cell_to_input_weights_ = { in SetUp()
1495 lstm.SetCellToInputWeights(cell_to_input_weights_); in TEST_F()
1558 lstm.SetCellToInputWeights(cell_to_input_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc70 cell_to_input_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel()
72 cell_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
77 cell_to_input_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel()
155 PopulateTensor(cell_to_input_weights_, f); in SetCellToInputWeights()
215 int cell_to_input_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
296 SetWeights(cell_to_input_weights_, f); in SetCellToInputWeights()
330 std::vector<float> cell_to_input_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
1390 cell_to_input_weights_ = { in SetUp()
1593 lstm.SetCellToInputWeights(cell_to_input_weights_); in TEST_F()
1660 lstm.SetCellToInputWeights(cell_to_input_weights_); in TEST_F()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2293 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
2295 cell_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
2300 cell_to_input_weights_ = AddNullInput(); in LSTMOpModel()
2374 SetData(cell_to_input_weights_, weight_type_, f); in SetCellToInputWeights()
2433 int cell_to_input_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2475 std::vector<float> cell_to_input_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
3181 cell_to_input_weights_ = { in SetUp()
3380 lstm.SetCellToInputWeights(cell_to_input_weights_); in TEST_F()