Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc208 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anonb3707f480111::LSTMOpModel
337 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anonb3707f480111::HybridLSTMOpModel
1499 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1562 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1626 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1731 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anonb3707f480111::HybridLayerNormLSTMOpModel
1940 layer_norm_lstm.SetProjectionWeights(projection_weights_); in TEST_F()
2036 layer_norm_lstm.SetProjectionWeights(projection_weights_); in TEST_F()
2131 layer_norm_lstm.SetProjectionWeights(projection_weights_); in TEST_F()
2267 layer_norm_lstm.SetProjectionWeights(projection_weights_); in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc182 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
307 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anon67ed54f90111::HybridUnidirectionalLSTMOpModel
1597 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1664 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1731 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
Doptional_tensor_test.cc173 void SetProjectionWeights(std::initializer_list<float> f) { in SetProjectionWeights() function in tflite::__anon67d22bc80111::LSTMOpModel
Dbidirectional_sequence_lstm_test.cc295 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anon821f96f80111::BidirectionalLSTMOpModel
1732 lstm.SetProjectionWeights( in TEST()
2435 lstm.SetProjectionWeights( in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2401 void SetProjectionWeights(const std::vector<float>& f) { in SetProjectionWeights() function in tflite::__anon0c68fcb70111::LSTMOpModel
3384 lstm.SetProjectionWeights(projection_weights_); in TEST_F()