Searched refs:InitDimsN (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1027 nvinfer1::Dims InitDimsN(std::initializer_list<int> list) { in InitDimsN() function
1059 InitDimsN({k, c, d, r, s}); // KCDRS shape (same as output) in ReorderDRSCKToKCDRS()
1062 InitDimsN({c * d * r * s, d * r * s, r * s, s, in ReorderDRSCKToKCDRS()
1066 InitDimsN({1, k, r * s * c * k, s * c * k, in ReorderDRSCKToKCDRS()
|