Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc172 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anonb3707f480111::LSTMOpModel
333 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
721 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
780 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
839 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1497 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1560 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1624 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1711 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anonb3707f480111::HybridLayerNormLSTMOpModel
1931 layer_norm_lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc162 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
303 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
807 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
870 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
932 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1595 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1662 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
1729 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
Doptional_tensor_test.cc153 void SetCellToOutputWeights(std::initializer_list<float> f) { in SetCellToOutputWeights() function in tflite::__anon67d22bc80111::LSTMOpModel
304 lstm.SetCellToOutputWeights( in TEST()
Dbidirectional_sequence_lstm_test.cc270 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
1014 lstm.SetCellToOutputWeights( in TEST()
1167 lstm.SetCellToOutputWeights( in TEST()
1726 lstm.SetCellToOutputWeights( in TEST()
2429 lstm.SetCellToOutputWeights( in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2381 void SetCellToOutputWeights(const std::vector<float>& f) { in SetCellToOutputWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel
2723 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()
3382 lstm.SetCellToOutputWeights(cell_to_output_weights_); in TEST_F()