Searched refs:input_to_forget_weights_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 55 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel() 137 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights() 231 int input_to_forget_weights_; member in tflite::__anonb3707f480111::LSTMOpModel 298 SetWeights(input_to_forget_weights_, f); in SetInputToForgetWeights() 347 std::vector<float> input_to_forget_weights_; member in tflite::__anonb3707f480111::BaseLstmTest 407 input_to_forget_weights_ = {0.09701663, 0.20334584, -0.50592935, in SetUp() 489 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 547 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 606 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 629 input_to_forget_weights_ = {-0.55291498, -0.42866567, 0.13056988, in SetUp() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 54 input_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel() 127 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights() 206 int input_to_forget_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 268 SetWeights(input_to_forget_weights_, f); in SetInputToForgetWeights() 320 std::vector<float> input_to_forget_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest 408 input_to_forget_weights_ = {0.09701663, 0.20334584, -0.50592935, in SetUp() 495 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 557 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 624 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 687 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() [all …]
|
D | optional_tensor_test.cc | 49 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 118 PopulateTensor(input_to_forget_weights_, f); in SetInputToForgetWeights() 199 int input_to_forget_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2277 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel() 2346 SetData(input_to_forget_weights_, weight_type_, f); in SetInputToForgetWeights() 2424 int input_to_forget_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2465 std::vector<float> input_to_forget_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2525 input_to_forget_weights_ = {0.09701663, 0.20334584, -0.50592935, in SetUp() 2608 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 2630 input_to_forget_weights_ = {-0.55291498, -0.42866567, 0.13056988, in SetUp() 2711 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F() 2752 input_to_forget_weights_ = { in SetUp() 3367 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
|