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()
|