Searched defs:squeeze_axis (Results 1 – 2 of 2) sorted by relevance
33 def squeeze_axis(shape, axis): function
678 std::vector<int> squeeze_axis{0}; in AdjustOtherGetItems() local802 …std::vector<int> squeeze_axis{1}; // our lstm output:0 have an extra axis that tflite not have, i… in Process() local