repalce toDimsPadAtEnd with toDimsTailPad

Signed-off-by: Ruoqian Guo <[email protected]>

Author: ruoqianguo

core/conversion/converters/impl/batch_norm.cpp

@@ -43,7 +43,7 @@ auto batch_norm_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns().
       auto should_unpack = util::toVec(orig_shape).size() < 4;
       if (should_unpack) {
         // expand spatial dims from 1D to 2D
-        auto new_shape = util::toDimsPadAtEnd(util::toVec(orig_shape), 4);
+        auto new_shape = util::toDimsTailPad(util::toVec(orig_shape), 4);
         LOG_DEBUG(
             "Input shape is less than 4D got: "
             << orig_shape << ", inserting shuffle layer to reshape to 4D tensor shape: " << new_shape);

core/util/trt_util.cpp

@@ -86,7 +86,7 @@ nvinfer1::Dims toDimsPad(c10::IntArrayRef l, uint64_t pad_to) {
   return dims;
 }
 
-nvinfer1::Dims toDimsPadAtEnd(c10::IntArrayRef l, uint64_t pad_to) {
+nvinfer1::Dims toDimsTailPad(c10::IntArrayRef l, uint64_t pad_to) {
   if (l.size() > pad_to) {
     LOG_DEBUG(
         "Requested padding of dimensions to " << pad_to << " but found " << l.size()
@@ -160,7 +160,7 @@ nvinfer1::Dims toDimsPad(c10::List<int64_t> l, uint64_t pad_to) {
   return dims;
 }
 
-nvinfer1::Dims toDimsPadAtEnd(c10::List<int64_t> l, uint64_t pad_to) {
+nvinfer1::Dims toDimsTailPad(c10::List<int64_t> l, uint64_t pad_to) {
   if (l.size() > pad_to) {
     LOG_DEBUG(
         "Requested padding of dimensions to " << pad_to << " but found " << l.size()

core/util/trt_util.h

@@ -92,8 +92,8 @@ int64_t volume(const nvinfer1::Dims& d);
 bool broadcastable(nvinfer1::Dims a, nvinfer1::Dims b, bool multidirectional = true);
 nvinfer1::Dims toDimsPad(c10::IntArrayRef l, uint64_t pad_to);
 nvinfer1::Dims toDimsPad(c10::List<int64_t> l, uint64_t pad_to);
-nvinfer1::Dims toDimsPadAtEnd(c10::IntArrayRef l, uint64_t pad_to);
-nvinfer1::Dims toDimsPadAtEnd(c10::List<int64_t> l, uint64_t pad_to);
+nvinfer1::Dims toDimsTailPad(c10::IntArrayRef l, uint64_t pad_to);
+nvinfer1::Dims toDimsTailPad(c10::List<int64_t> l, uint64_t pad_to);
 nvinfer1::Dims unpadDims(const nvinfer1::Dims& d);
 nvinfer1::Dims unsqueezeDims(const nvinfer1::Dims& d, int pos);
 nvinfer1::Dims squeezeDims(const nvinfer1::Dims& d, int pos);