Searched refs:relu6_layer (Results 1 – 1 of 1) sorted by relevance
4055 nvinfer1::IElementWiseLayer* relu6_layer = in ConvertRelu6() local4059 TFTRT_RETURN_ERROR_IF_NULLPTR(relu6_layer, node_def.name()); in ConvertRelu6()4060 params->converter->SetLayerName(relu6_layer, node_def, "min"); in ConvertRelu6()4061 nvinfer1::ITensor* output_tensor = relu6_layer->getOutput(0); in ConvertRelu6()