Searched refs:tf_stride (Results 1 – 1 of 1) sorted by relevance
2320 const auto tf_stride = attrs.get<std::vector<int64>>("strides"); in ConvertConv2DHelper() local2321 if (tf_stride.size() != 4) { in ConvertConv2DHelper()2326 if (tf_stride[0] != 1 || tf_stride[c_index] != 1) { in ConvertConv2DHelper()2337 const nvinfer1::DimsHW stride(tf_stride[h_index], tf_stride[w_index]); in ConvertConv2DHelper()3282 const auto tf_stride = attrs.get<std::vector<int64>>("strides"); in ConvertConv3DHelper() local3283 if (tf_stride.size() != kNumDims) { in ConvertConv3DHelper()3288 if (tf_stride[0] != 1 || tf_stride[c_index] != 1) { in ConvertConv3DHelper()3294 const nvinfer1::Dims3 stride_dhw(tf_stride[d_index], tf_stride[h_index], in ConvertConv3DHelper()3295 tf_stride[w_index]); in ConvertConv3DHelper()3444 const auto tf_stride = attrs.get<std::vector<int64>>("strides"); in ConvertPool3D() local[all …]