Home
last modified time | relevance | path

Searched refs:SetInputToForgetWeights (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc136 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anonb3707f480111::LSTMOpModel
297 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
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()
709 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
768 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
827 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
1482 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
1545 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc126 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
267 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
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()
795 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
858 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
920 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
1580 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
[all …]
Dbidirectional_sequence_lstm_test.cc225 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
496 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, in TEST_P()
661 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, in TEST_P()
825 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, in TEST()
982 lstm.SetInputToForgetWeights({-0.55291498, -0.42866567, 0.13056988, in TEST()
1135 lstm.SetInputToForgetWeights({-0.55291498, -0.42866567, 0.13056988, in TEST()
1306 lstm.SetInputToForgetWeights( in TEST()
2009 lstm.SetInputToForgetWeights( in TEST()
2702 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, in TEST_P()
Doptional_tensor_test.cc117 void SetInputToForgetWeights(std::initializer_list<float> f) { in SetInputToForgetWeights() function in tflite::__anon67d22bc80111::LSTMOpModel
272 lstm.SetInputToForgetWeights({-0.55291498, -0.42866567, 0.13056988, in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2345 void SetInputToForgetWeights(const std::vector<float>& f) { in SetInputToForgetWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel
2608 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
2711 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
3367 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()