Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm_test.cc138 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
279 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
503 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
565 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
632 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
695 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1588 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1655 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1722 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
Dlstm_test.cc148 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anonb3707f480111::LSTMOpModel
309 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
497 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
555 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
614 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1490 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1553 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1617 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
1687 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anonb3707f480111::HybridLayerNormLSTMOpModel
1924 layer_norm_lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
[all …]
Dbidirectional_sequence_lstm_test.cc240 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
512 lstm.SetRecurrentToInputWeights( in TEST_P()
677 lstm.SetRecurrentToInputWeights( in TEST_P()
841 lstm.SetRecurrentToInputWeights( in TEST()
1401 lstm.SetRecurrentToInputWeights( in TEST()
2104 lstm.SetRecurrentToInputWeights( in TEST()
2718 lstm.SetRecurrentToInputWeights( in TEST_P()
Doptional_tensor_test.cc129 void SetRecurrentToInputWeights(std::initializer_list<float> f) { in SetRecurrentToInputWeights() function in tflite::__anon67d22bc80111::LSTMOpModel
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2357 void SetRecurrentToInputWeights(const std::vector<float>& f) { in SetRecurrentToInputWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel
2616 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()
3375 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights_); in TEST_F()