Searched refs:mul_layer (Results 1 – 1 of 1) sorted by relevance
2756 nvinfer1::IElementWiseLayer* mul_layer = in ConvertLeakyRelu() local2761 TFTRT_RETURN_ERROR_IF_NULLPTR(mul_layer, node_def.name()); in ConvertLeakyRelu()2766 *const_cast<nvinfer1::ITensor*>(mul_layer->getOutput(0)), in ConvertLeakyRelu()2771 output_tensor, const_cast<nvinfer1::ITensor*>(mul_layer->getOutput(0))); in ConvertLeakyRelu()