Searched refs:istrides (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 769 const nvinfer1::DimsNCHW& istrides, T* odata, in Reorder4() argument
776 w * ostrides.w()] = idata[n * istrides.n() + c * istrides.c() + in Reorder4()
777 h * istrides.h() + w * istrides.w()]; in Reorder4()
786 const nvinfer1::DimsHW& istrides, T* odata, in Reorder2() argument
791 idata[h * istrides.h() + w * istrides.w()]; in Reorder2()
803 const nvinfer1::DimsHW istrides = {1, k}; in ReorderCKtoKC() local
808 istrides, in ReorderCKtoKC()
817 istrides, in ReorderCKtoKC()
847 const nvinfer1::DimsNCHW istrides = {1, k, s * k * c, c * k}; in ReorderRSCKToKCRS() local
852 istrides, in ReorderRSCKToKCRS()
[all …]
|