Searched refs:Reorder4 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 931 void Reorder4(const nvinfer1::DimsNCHW& shape, const T* idata, in Reorder4() function
1009 Reorder4({k, c, r, s}, static_cast<float const*>(iweights.GetValues()), in ReorderRSCKToKCRS()
1014 Reorder4({k, c, r, s}, in ReorderRSCKToKCRS()
|