Searched refs:ostrides (Results 1 – 1 of 1) sorted by relevance
770 const nvinfer1::DimsNCHW& ostrides) { in Reorder4() argument775 odata[n * ostrides.n() + c * ostrides.c() + h * ostrides.h() + in Reorder4()776 w * ostrides.w()] = idata[n * istrides.n() + c * istrides.c() + in Reorder4()787 const nvinfer1::DimsHW& ostrides) { in Reorder2() argument790 odata[h * ostrides.h() + w * ostrides.w()] = in Reorder2()804 const nvinfer1::DimsHW ostrides = {c, 1}; in ReorderCKtoKC() local811 ostrides); in ReorderCKtoKC()819 ostrides); in ReorderCKtoKC()848 const nvinfer1::DimsNCHW ostrides = {c * r * s, r * s, s, 1}; in ReorderRSCKToKCRS() local854 ostrides); in ReorderRSCKToKCRS()[all …]