Home
last modified time | relevance | path

Searched refs:projection_weights_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc93 projection_weights_ = AddInput(weight_type); in LSTMOpModel()
100 projection_weights_ = AddNullInput(); in LSTMOpModel()
209 PopulateTensor(projection_weights_, f); in SetProjectionWeights()
254 int projection_weights_; member in tflite::__anonb3707f480111::LSTMOpModel
338 SetWeights(projection_weights_, f); in SetProjectionWeights()
360 std::vector<float> projection_weights_; member in tflite::__anonb3707f480111::BaseLstmTest
1315 projection_weights_ = { in SetUp()
1499 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1562 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1626 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
[all …]
Doptional_tensor_test.cc87 projection_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
94 projection_weights_ = AddNullInput(); in LSTMOpModel()
174 PopulateTensor(projection_weights_, f); in SetProjectionWeights()
217 int projection_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
Dunidirectional_sequence_lstm_test.cc92 projection_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
99 projection_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel()
183 PopulateTensor(projection_weights_, f); in SetProjectionWeights()
224 int projection_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
308 SetWeights(projection_weights_, f); in SetProjectionWeights()
333 std::vector<float> projection_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest
1408 projection_weights_ = { in SetUp()
1597 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1664 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
1731 lstm.SetProjectionWeights(projection_weights_); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2315 projection_weights_ = AddInput(weight_type); in LSTMOpModel()
2322 projection_weights_ = AddNullInput(); in LSTMOpModel()
2402 SetData(projection_weights_, weight_type_, f); in SetProjectionWeights()
2442 int projection_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel
2478 std::vector<float> projection_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest
3199 projection_weights_ = { in SetUp()
3384 lstm.SetProjectionWeights(projection_weights_); in TEST_F()