Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1794 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertConv2DHelper() local
1795 if (tf_dilations.size() != 4) { in ConvertConv2DHelper()
1800 if (tf_dilations[0] != 1 || tf_dilations[c_index] != 1) { in ConvertConv2DHelper()
1805 const nvinfer1::DimsHW dilation(tf_dilations[h_index], tf_dilations[w_index]); in ConvertConv2DHelper()