Searched defs:axis_in (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | reduce_utils_test.cc | 25 void TestFunction(const std::vector<int>& axis_in, in TestFunction()
48 const std::vector<int> axis_in = {0, 1, 2}; in TEST() local
56 const std::vector<int> axis_in = {3, 2}; in TEST() local
64 const std::vector<int> axis_in = {}; in TEST() local
72 const std::vector<int> axis_in = {0}; in TEST() local
80 const std::vector<int> axis_in = {-2}; in TEST() local
88 const std::vector<int> axis_in = {-1}; in TEST() local
96 const std::vector<int> axis_in = {2, 1, 2, 1, 2, 1}; in TEST() local
104 const std::vector<int> axis_in = {2, -1, -2, -1, 2, 1}; in TEST() local
112 const std::vector<int> axis_in = {0}; in TEST() local
[all …]
|