Searched refs:pad_layer (Results 1 – 1 of 1) sorted by relevance
1885 auto pad_layer = params->converter->network()->addPadding( in ConvertConv2DHelper() local1889 TFTRT_RETURN_ERROR_IF_NULLPTR(pad_layer, node_def.name()); in ConvertConv2DHelper()1891 const_cast<nvinfer1::ITensor*>(tensor), pad_layer->getOutput(0)); in ConvertConv2DHelper()1893 tensor = pad_layer->getOutput(0); in ConvertConv2DHelper()2696 auto pad_layer = params->converter->network()->addPadding( in ConvertPool() local2700 TFTRT_RETURN_ERROR_IF_NULLPTR(pad_layer, node_def.name()); in ConvertPool()2702 const_cast<nvinfer1::ITensor*>(tensor), pad_layer->getOutput(0)); in ConvertPool()2704 tensor = pad_layer->getOutput(0); in ConvertPool()