Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval_test.cc58 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::__anonb312a86f0111::BaseLstmParam
[all …]