Home
last modified time | relevance | path

Searched refs:ostrides (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc770 const nvinfer1::DimsNCHW& ostrides) { in Reorder4() argument
775 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() argument
790 odata[h * ostrides.h() + w * ostrides.w()] = in Reorder2()
804 const nvinfer1::DimsHW ostrides = {c, 1}; in ReorderCKtoKC() local
811 ostrides); in ReorderCKtoKC()
819 ostrides); in ReorderCKtoKC()
848 const nvinfer1::DimsNCHW ostrides = {c * r * s, r * s, s, 1}; in ReorderRSCKToKCRS() local
854 ostrides); in ReorderRSCKToKCRS()
[all …]