Searched refs:dilation_dhw (Results 1 – 1 of 1) sorted by relevance
3271 const nvinfer1::Dims3 dilation_dhw( in ConvertConv3DHelper() local3274 (dilation_dhw.d[0] != 1 || dilation_dhw.d[1] != 1 || in ConvertConv3DHelper()3275 dilation_dhw.d[2] != 1)) { in ConvertConv3DHelper()3305 (weights.shape_.d[0] - 1) * (dilation_dhw.d[0] - 1), // D in ConvertConv3DHelper()3307 (weights.shape_.d[1] - 1) * (dilation_dhw.d[1] - 1), // R in ConvertConv3DHelper()3309 (weights.shape_.d[2] - 1) * (dilation_dhw.d[2] - 1) // S in ConvertConv3DHelper()3393 layer->setDilationNd(dilation_dhw); in ConvertConv3DHelper()