Searched refs:trt_output_shape (Results 1 – 1 of 1) sorted by relevance
2440 nvinfer1::Dims trt_output_shape = output_tensor->getDimensions(); in ConvertConv2DHelper() local2443 const int height_diff = output_height - trt_output_shape.d[1]; in ConvertConv2DHelper()2444 const int width_diff = output_width - trt_output_shape.d[2]; in ConvertConv2DHelper()2452 trt_output_shape.d[1], ", ", trt_output_shape.d[2], ") for op ", in ConvertConv2DHelper()