Searched refs:tf_stride (Results 1 – 1 of 1) sorted by relevance
1812 const auto tf_stride = attrs.get<std::vector<int64>>("strides"); in ConvertConv2DHelper() local1813 if (tf_stride.size() != 4) { in ConvertConv2DHelper()1818 if (tf_stride[0] != 1 || tf_stride[c_index] != 1) { in ConvertConv2DHelper()1823 const nvinfer1::DimsHW stride(tf_stride[h_index], tf_stride[w_index]); in ConvertConv2DHelper()2673 const auto tf_stride = attrs.get<std::vector<int64>>("strides"); in ConvertPool() local2674 const nvinfer1::DimsHW stride(tf_stride[h_index], tf_stride[w_index]); in ConvertPool()