Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2302 auto tf_dilations = attrs.get<std::vector<int64>>("dilations"); in ConvertConv2DHelper() local
2303 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() local
3260 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() local
3536 if (tf_dilations.size() != 4) { in ConvertFusedConv2DBiasActivation()
[all …]
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py229 tf_dilations = list(self.dilation_rate)
240 dilations=tf_dilations,