Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc76 cell_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
80 cell_to_output_weights_ = AddNullInput(); in LSTMOpModel()
173 PopulateTensor(cell_to_output_weights_, f); in SetCellToOutputWeights()
242 int cell_to_output_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
334 SetWeights(cell_to_output_weights_, f); in SetCellToOutputWeights()
359 std::vector<float> cell_to_output_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
660 cell_to_output_weights_ = {-0.17135078, 0.82760304, 0.85573703, in SetUp()
721 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
780 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
839 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc75 cell_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
79 cell_to_output_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel()
163 PopulateTensor(cell_to_output_weights_, f); in SetCellToOutputWeights()
217 int cell_to_output_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
304 SetWeights(cell_to_output_weights_, f); in SetCellToOutputWeights()
332 std::vector<float> cell_to_output_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
741 cell_to_output_weights_ = {-0.17135078, 0.82760304, 0.85573703, in SetUp()
807 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
870 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
932 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc70 cell_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
74 cell_to_output_weights_ = AddNullInput(); in LSTMOpModel()
154 PopulateTensor(cell_to_output_weights_, f); in SetCellToOutputWeights()
210 int cell_to_output_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2298 cell_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
2302 cell_to_output_weights_ = AddNullInput(); in LSTMOpModel()
2382 SetData(cell_to_output_weights_, weight_type_, f); in SetCellToOutputWeights()
2435 int cell_to_output_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2477 std::vector<float> cell_to_output_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
2661 cell_to_output_weights_ = {-0.17135078, 0.82760304, 0.85573703, in SetUp()
2723 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
3193 cell_to_output_weights_ = { in SetUp()
3382 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()