Searched refs:input_to_cell_weights_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 56 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel() 141 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights() 232 int input_to_cell_weights_; member in tflite::__anonb3707f480111::LSTMOpModel 302 SetWeights(input_to_cell_weights_, f); in SetInputToCellWeights() 346 std::vector<float> input_to_cell_weights_; member in tflite::__anonb3707f480111::BaseLstmTest 405 input_to_cell_weights_ = {-0.50013041, 0.1370284, 0.11810488, 0.2013163, in SetUp() 488 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 546 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 605 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 625 input_to_cell_weights_ = {-0.49770179, -0.27711356, -0.09624726, in SetUp() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 55 input_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel() 131 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights() 207 int input_to_cell_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 272 SetWeights(input_to_cell_weights_, f); in SetInputToCellWeights() 319 std::vector<float> input_to_cell_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest 406 input_to_cell_weights_ = {-0.50013041, 0.1370284, 0.11810488, 0.2013163, in SetUp() 494 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 556 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 623 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 686 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() [all …]
|
D | optional_tensor_test.cc | 50 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 122 PopulateTensor(input_to_cell_weights_, f); in SetInputToCellWeights() 200 int input_to_cell_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2278 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel() 2350 SetData(input_to_cell_weights_, weight_type_, f); in SetInputToCellWeights() 2425 int input_to_cell_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2464 std::vector<float> input_to_cell_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2523 input_to_cell_weights_ = {-0.50013041, 0.1370284, 0.11810488, 0.2013163, in SetUp() 2607 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 2626 input_to_cell_weights_ = {-0.49770179, -0.27711356, -0.09624726, in SetUp() 2710 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F() 2779 input_to_cell_weights_ = { in SetUp() 3366 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F()
|