Searched refs:trt_pad_index (Results 1 – 1 of 1) sorted by relevance
4897 std::vector<int> trt_pad_index{nb_dims - 2, nb_dims - 1}; in ConvertPad() local4910 trt_pad_index[0] = nb_dims - 1; in ConvertPad()4924 std::swap(trt_pad_index[0], trt_pad_index[1]); in ConvertPad()4930 const int trt_index = trt_pad_index[i]; in ConvertPad()