Searched refs:SetForgetGateBias (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 196 void SetForgetGateBias(const std::vector<float>& f) { in SetForgetGateBias() function in tflite::__anonb3707f480111::LSTMOpModel 494 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 552 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 611 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 713 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 772 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 831 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 1487 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 1550 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 1614 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 170 void SetForgetGateBias(const std::vector<float>& f) { in SetForgetGateBias() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 500 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 562 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 629 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 692 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 799 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 862 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 924 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 1585 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 1652 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() [all …]
|
D | bidirectional_sequence_lstm_test.cc | 280 void SetForgetGateBias(const std::vector<float>& f) { in SetForgetGateBias() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel 508 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST_P() 673 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST_P() 837 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST() 992 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST() 1145 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST() 1383 lstm.SetForgetGateBias({0.035185695, -0.042891346, -0.03032477, 0.23027696, in TEST() 2086 lstm.SetForgetGateBias({0.035185695, -0.042891346, -0.03032477, 0.23027696, in TEST() 2714 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST_P()
|
D | optional_tensor_test.cc | 161 void SetForgetGateBias(std::initializer_list<float> f) { in SetForgetGateBias() function in tflite::__anon67d22bc80111::LSTMOpModel 282 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2389 void SetForgetGateBias(const std::vector<float>& f) { in SetForgetGateBias() function in tflite::__anon0c68fcb70111::LSTMOpModel 2613 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 2715 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F() 3372 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
|