Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc168 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anonb3707f480111::LSTMOpModel
329 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
720 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
779 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
838 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1496 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1559 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1623 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1707 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anonb3707f480111::HybridLayerNormLSTMOpModel
1930 layer_norm_lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc158 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
299 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
806 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
869 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
931 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1594 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1661 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
1728 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
Doptional_tensor_test.cc149 void SetCellToForgetWeights(std::initializer_list<float> f) { in SetCellToForgetWeights() function in tflite::__anon67d22bc80111::LSTMOpModel
302 lstm.SetCellToForgetWeights( in TEST()
Dbidirectional_sequence_lstm_test.cc265 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
1012 lstm.SetCellToForgetWeights( in TEST()
1165 lstm.SetCellToForgetWeights( in TEST()
1720 lstm.SetCellToForgetWeights( in TEST()
2423 lstm.SetCellToForgetWeights( in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2377 void SetCellToForgetWeights(const std::vector<float>& f) { in SetCellToForgetWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel
2722 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()
3381 lstm.SetCellToForgetWeights(cell_to_forget_weights_); in TEST_F()