Searched refs:stride_dhw (Results 1 – 1 of 1) sorted by relevance
3294 const nvinfer1::Dims3 stride_dhw(tf_stride[d_index], tf_stride[h_index], in ConvertConv3DHelper() local3319 CreateSamePadding(stride_dhw, effective_kernel_size, input_dims); in ConvertConv3DHelper()3368 layer->setStrideNd(stride_dhw); // change to nd set stride in ConvertConv3DHelper()3385 layer->setStrideNd(stride_dhw); in ConvertConv3DHelper()