Searched defs:istrides (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| H A D | convert_nodes.cc | 271 nvinfer1::DimsNCHW istrides, T* odata,
278 w * ostrides.w()] = idata[n * istrides.n() + c * istrides.c() +
279 h * istrides.h() + w * istrides.w()];
298 nvinfer1::DimsNCHW istrides = {1, k, s * k * c, c * k}; local
303 istrides,
270 Reorder4(nvinfer1::DimsNCHW shape, const T* idata, nvinfer1::DimsNCHW istrides, T* odata, nvinfer1::DimsNCHW ostrides) argument
|
Completed in 112 milliseconds