Searched refs:SetCellBias (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 200 void SetCellBias(const std::vector<float>& f) { in SetCellBias() function in tflite::__anonb3707f480111::LSTMOpModel 493 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 551 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 610 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 712 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 771 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 830 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 1486 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 1549 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 1613 lstm.SetCellBias(cell_gate_bias_); in TEST_F() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 174 void SetCellBias(const std::vector<float>& f) { in SetCellBias() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 499 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 561 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 628 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 691 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 798 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 861 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 923 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 1584 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 1651 lstm.SetCellBias(cell_gate_bias_); in TEST_F() [all …]
|
D | bidirectional_sequence_lstm_test.cc | 285 void SetCellBias(const std::vector<float>& f) { in SetCellBias() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel 506 lstm.SetCellBias({0., 0., 0., 0.}); in TEST_P() 671 lstm.SetCellBias({0., 0., 0., 0.}); in TEST_P() 835 lstm.SetCellBias({0., 0., 0., 0.}); in TEST() 990 lstm.SetCellBias({0., 0., 0., 0.}); in TEST() 1143 lstm.SetCellBias({0., 0., 0., 0.}); in TEST() 1389 lstm.SetCellBias({-0.024379363, 0.0055531194, 0.23377132, 0.033463873, in TEST() 2092 lstm.SetCellBias({-0.024379363, 0.0055531194, 0.23377132, 0.033463873, in TEST() 2712 lstm.SetCellBias({0., 0., 0., 0.}); in TEST_P()
|
D | optional_tensor_test.cc | 165 void SetCellBias(std::initializer_list<float> f) { in SetCellBias() function in tflite::__anon67d22bc80111::LSTMOpModel 280 lstm.SetCellBias({0., 0., 0., 0.}); in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2393 void SetCellBias(const std::vector<float>& f) { in SetCellBias() function in tflite::__anon0c68fcb70111::LSTMOpModel 2612 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 2714 lstm.SetCellBias(cell_gate_bias_); in TEST_F() 3371 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
|