Searched refs:SetInputToOutputWeights (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 144 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anonb3707f480111::LSTMOpModel 305 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel 490 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 548 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 607 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 710 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 769 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 828 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 1483 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 1546 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 134 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 275 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel 496 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 558 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 625 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 688 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 796 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 859 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 921 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 1581 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() [all …]
|
D | bidirectional_sequence_lstm_test.cc | 235 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel 500 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, in TEST_P() 665 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, in TEST_P() 829 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, in TEST() 986 lstm.SetInputToOutputWeights({0.10725588, -0.02335852, -0.55932593, in TEST() 1139 lstm.SetInputToOutputWeights({0.10725588, -0.02335852, -0.55932593, in TEST() 1355 lstm.SetInputToOutputWeights( in TEST() 2058 lstm.SetInputToOutputWeights( in TEST() 2706 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, in TEST_P()
|
D | optional_tensor_test.cc | 125 void SetInputToOutputWeights(std::initializer_list<float> f) { in SetInputToOutputWeights() function in tflite::__anon67d22bc80111::LSTMOpModel 276 lstm.SetInputToOutputWeights({0.10725588, -0.02335852, -0.55932593, in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2353 void SetInputToOutputWeights(const std::vector<float>& f) { in SetInputToOutputWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel 2609 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 2712 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 3368 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F()
|