Searched refs:ostrides (Results 1 – 1 of 1) sorted by relevance
910 const nvinfer1::Dims& ostrides) { in Reorder5() argument916 odata[k * ostrides.d[0] + c * ostrides.d[1] + d * ostrides.d[2] + in Reorder5()917 r * ostrides.d[3] + s * ostrides.d[4]] = in Reorder5()933 const nvinfer1::DimsNCHW& ostrides) { in Reorder4() argument938 odata[n * ostrides.n() + c * ostrides.c() + h * ostrides.h() + in Reorder4()939 w * ostrides.w()] = idata[n * istrides.n() + c * istrides.c() + in Reorder4()950 const nvinfer1::DimsHW& ostrides) { in Reorder2() argument953 odata[h * ostrides.h() + w * ostrides.w()] = in Reorder2()967 const nvinfer1::DimsHW ostrides = {c, 1}; in ReorderCKtoKC() local971 istrides, static_cast<float*>(oweights->GetValues()), ostrides); in ReorderCKtoKC()[all …]