Searched refs:tf_dilations (Results 1 – 2 of 2) sorted by relevance
2302 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertConv2DHelper() local2303 if (tf_dilations.size() != 4) { in ConvertConv2DHelper()2308 if (tf_dilations[0] != 1 || tf_dilations[c_index] != 1) { in ConvertConv2DHelper()2313 const nvinfer1::DimsHW dilation(tf_dilations[h_index], tf_dilations[w_index]); in ConvertConv2DHelper()3259 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertConv3DHelper() local3260 if (tf_dilations.size() != kNumDims) { in ConvertConv3DHelper()3265 if (tf_dilations[0] != 1 || tf_dilations[c_index] != 1) { in ConvertConv3DHelper()3272 tf_dilations[d_index], tf_dilations[h_index], tf_dilations[w_index]); in ConvertConv3DHelper()3535 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertFusedConv2DBiasActivation() local3536 if (tf_dilations.size() != 4) { in ConvertFusedConv2DBiasActivation()[all …]
229 tf_dilations = list(self.dilation_rate)240 dilations=tf_dilations,