Searched refs:trt_axes (Results 1 – 1 of 1) sorted by relevance
2814 std::vector<int> trt_axes; in ConvertSqueeze() local2815 trt_axes.reserve(squeeze_dims.size()); in ConvertSqueeze()2829 trt_axes.push_back(trt_axis); in ConvertSqueeze()2832 for (int axis : trt_axes) { in ConvertSqueeze()