Searched refs:MakeLiteTensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | buffer_map_test.cc | 43 UniqueTfLiteTensor MakeLiteTensor(const std::vector<int>& shape, in MakeLiteTensor() function
59 UniqueTfLiteTensor MakeLiteTensor<string>(const std::vector<int>& shape, in MakeLiteTensor() function
111 MakeLiteTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
129 MakeLiteTensor<string>({1, 2, 1, 3}, {"", "", "", "str1", "", ""}); in TEST()
145 MakeLiteTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
147 MakeLiteTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}); in TEST()
159 MakeLiteTensor<float>({1, 2, 1, 3}, {0, 0, 0, 0.123f, 0, 0}); in TEST()
161 MakeLiteTensor<string>({1, 2, 4}, {"", "", "", "s3", "", "", "s1", "s2"}); in TEST()
231 MakeLiteTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}); in TEST()
246 MakeLiteTensor<int>({1, 2, 4}, {0, 0, 0, 3, 0, 0, 1, 2}); in TEST()
|