Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm_test.cc166 void SetInputGateBias(const std::vector<float>& f) { in SetInputGateBias() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
498 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
560 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
627 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
690 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1583 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1650 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1717 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
Dlstm_test.cc192 void SetInputGateBias(const std::vector<float>& f) { in SetInputGateBias() function in tflite::__anonb3707f480111::LSTMOpModel
492 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
550 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
609 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1485 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1548 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1612 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
1919 layer_norm_lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
2015 layer_norm_lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
2110 layer_norm_lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
Dbidirectional_sequence_lstm_test.cc275 void SetInputGateBias(const std::vector<float>& f) { in SetInputGateBias() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
504 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST_P()
669 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST_P()
833 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST()
1377 lstm.SetInputGateBias( in TEST()
2080 lstm.SetInputGateBias( in TEST()
2710 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST_P()
Doptional_tensor_test.cc157 void SetInputGateBias(std::initializer_list<float> f) { in SetInputGateBias() function in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2385 void SetInputGateBias(const std::vector<float>& f) { in SetInputGateBias() function in tflite::__anon0c68fcb70111::LSTMOpModel
2611 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
3370 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()