Searched refs:tf_dilations (Results 1 – 1 of 1) sorted by relevance
1794 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertConv2DHelper() local1795 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()