Searched refs:PackWeightToTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval_test.cc | 58 PackWeightToTensor(&i2i_tensor_, i2i_, i2i_size_); in Geti2i()
63 PackWeightToTensor(&i2f_tensor_, i2f_, i2f_size_); in Geti2f()
68 PackWeightToTensor(&i2c_tensor_, i2c_, i2c_size_); in Geti2c()
73 PackWeightToTensor(&i2o_tensor_, i2o_, i2o_size_); in Geti2o()
78 PackWeightToTensor(&r2i_tensor_, r2i_, r2i_size_); in Getr2i()
83 PackWeightToTensor(&r2f_tensor_, r2f_, r2f_size_); in Getr2f()
88 PackWeightToTensor(&r2c_tensor_, r2c_, r2c_size_); in Getr2c()
93 PackWeightToTensor(&r2o_tensor_, r2o_, r2o_size_); in Getr2o()
98 PackWeightToTensor(&projection_tensor_, projection_, projection_size_); in GetProjection()
129 void PackWeightToTensor(TfLiteTensor* tensor, std::vector<T>& data, in PackWeightToTensor() function in tflite::__anon39d466b60111::BaseLstmParam
[all …]
|