Searched refs:mul_layer (Results 1 – 1 of 1) sorted by relevance
3819 nvinfer1::IElementWiseLayer* mul_layer = in ConvertLeakyRelu() local3822 TFTRT_RETURN_ERROR_IF_NULLPTR(mul_layer, node_def.name()); in ConvertLeakyRelu()3823 params->converter->SetLayerName(mul_layer, node_def, "mul"); in ConvertLeakyRelu()3827 *tensor, *mul_layer->getOutput(0), in ConvertLeakyRelu()3830 params->converter->SetLayerName(mul_layer, node_def, "max"); in ConvertLeakyRelu()3833 output_tensor, mul_layer->getOutput(0)); in ConvertLeakyRelu()