Searched refs:size_after_squeeze (Results 1 – 1 of 1) sorted by relevance
5002 std::vector<int> size_after_squeeze(size); in ConvertSplitHelper() local5003 size_after_squeeze.erase(size_after_squeeze.begin() + trt_axis + 1); in ConvertSplitHelper()5005 size_after_squeeze, &final_shape_for_unpack, /*ignore_frst_dim=*/true)); in ConvertSplitHelper()