Searched refs:SetAxis (Results 1 – 5 of 5) sorted by relevance
42 void SetAxis(int axis) { PopulateTensor<int32_t>(axis_, {axis}); } in SetAxis() function in tflite::__anon558ad8b40111::ExpandDimsOpModel56 m.SetAxis(0); in TEST()61 m.SetAxis(1); in TEST()66 m.SetAxis(2); in TEST()71 m.SetAxis(-1); in TEST()
29 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis() function in tflite::__anon5bc6b6600111::BaseOpModel338 m.SetAxis(axis); in TEST()353 m.SetAxis(axis); in TEST()366 m.SetAxis(axis); in TEST()441 m.SetAxis(axis); in TEST()457 m.SetAxis(axis); in TEST()529 m.SetAxis(axis); in TEST()554 m.SetAxis(axis); in TEST()567 m.SetAxis(axis); in TEST()620 m.SetAxis(axis); in TEST()[all …]
54 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis() function in tflite::__anon0e1332340111::SplitOpModel82 m.SetAxis(axis); in Check()
60 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis() function in tflite::__anon6dfc68290111::SplitVOpModel87 m.SetAxis(axis); in Check()
3391 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis() function in tflite::__anon0c68fcb70111::BaseReduceOpModel