Searched refs:relu6_layer (Results 1 – 1 of 1) sorted by relevance
2912 nvinfer1::IElementWiseLayer* relu6_layer = in ConvertRelu6() local2917 TFTRT_RETURN_ERROR_IF_NULLPTR(relu6_layer, node_def.name()); in ConvertRelu6()2918 nvinfer1::ITensor* output_tensor = relu6_layer->getOutput(0); in ConvertRelu6()