Searched refs:SetOutputGateBias (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 204 void SetOutputGateBias(const std::vector<float>& f) { in SetOutputGateBias() function in tflite::__anonb3707f480111::LSTMOpModel 495 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 553 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 612 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 714 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 773 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 832 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 1488 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 1551 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 1615 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 178 void SetOutputGateBias(const std::vector<float>& f) { in SetOutputGateBias() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 501 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 563 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 630 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 693 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 800 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 863 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 925 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 1586 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 1653 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() [all …]
|
D | bidirectional_sequence_lstm_test.cc | 290 void SetOutputGateBias(const std::vector<float>& f) { in SetOutputGateBias() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel 510 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST_P() 675 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST_P() 839 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST() 994 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST() 1147 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST() 1395 lstm.SetOutputGateBias( in TEST() 2098 lstm.SetOutputGateBias( in TEST() 2716 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST_P()
|
D | optional_tensor_test.cc | 169 void SetOutputGateBias(std::initializer_list<float> f) { in SetOutputGateBias() function in tflite::__anon67d22bc80111::LSTMOpModel 284 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2397 void SetOutputGateBias(const std::vector<float>& f) { in SetOutputGateBias() function in tflite::__anon0c68fcb70111::LSTMOpModel 2614 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 2716 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 3373 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
|