Searched refs:SetAxis (Results 1 – 5 of 5) sorted by relevance
33 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis() function in tflite::__anon5bc6b6600111::BaseOpModel390 m.SetAxis(axis); in TEST()405 m.SetAxis(axis); in TEST()423 m.SetAxis(axis); in TEST()436 m.SetAxis(axis); in TEST()599 m.SetAxis(axis); in TEST()615 m.SetAxis(axis); in TEST()697 m.SetAxis(axis); in TEST()732 m.SetAxis(axis); in TEST()745 m.SetAxis(axis); in TEST()[all …]
64 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis() function in tflite::__anon6dfc68290111::SplitVOpModel92 m.SetAxis(axis); in Check()
65 void SetAxis(int axis) { PopulateTensor(axis_, {axis}); } in SetAxis() function in tflite::__anon0e1332340111::SplitOpModel95 m.SetAxis(axis); in Check()
276 RETURN_IF_ERROR(SetAxis(input_shapes, &attr.axis)); in Parse()302 absl::Status SetAxis(const std::vector<BHWC>& input_shapes, Axis* axis) { in SetAxis() function in tflite::gpu::__anon48bc1b4d0111::ConcatenationOperationParser
4654 void SetAxis(const std::vector<int>& data) { PopulateTensor(axis_, data); } in SetAxis() function in tflite::__anon0c68fcb70111::BaseReduceOpModel4703 m.SetAxis(axis); in TEST()