Searched refs:istrides (Results 1 – 1 of 1) sorted by relevance
909 const nvinfer1::Dims& istrides, T* odata, in Reorder5() argument918 idata[k * istrides.d[0] + c * istrides.d[1] + in Reorder5()919 d * istrides.d[2] + r * istrides.d[3] + in Reorder5()920 s * istrides.d[4]]; in Reorder5()932 const nvinfer1::DimsNCHW& istrides, T* odata, in Reorder4() argument939 w * ostrides.w()] = idata[n * istrides.n() + c * istrides.c() + in Reorder4()940 h * istrides.h() + w * istrides.w()]; in Reorder4()949 const nvinfer1::DimsHW& istrides, T* odata, in Reorder2() argument954 idata[h * istrides.h() + w * istrides.w()]; in Reorder2()966 const nvinfer1::DimsHW istrides = {1, k}; in ReorderCKtoKC() local[all …]