Searched refs:conv_layer (Results 1 – 1 of 1) sorted by relevance
2392 nvinfer1::ILayer* conv_layer = nullptr; in ConvertConv2DHelper() local2411 conv_layer = layer; in ConvertConv2DHelper()2428 conv_layer = layer; in ConvertConv2DHelper()2430 params->converter->SetLayerName(conv_layer, node_def, "conv"); in ConvertConv2DHelper()2431 nvinfer1::ITensor* output_tensor = conv_layer->getOutput(0); in ConvertConv2DHelper()3361 nvinfer1::ILayer* conv_layer = nullptr; in ConvertConv3DHelper() local3378 conv_layer = layer; in ConvertConv3DHelper()3394 conv_layer = layer; in ConvertConv3DHelper()3396 params->converter->SetLayerName(conv_layer, node_def, "conv"); in ConvertConv3DHelper()3397 nvinfer1::ITensor* output_tensor = conv_layer->getOutput(0); in ConvertConv3DHelper()[all …]