Searched refs:sqrt_layer (Results 1 – 1 of 1) sorted by relevance
3221 nvinfer1::IUnaryLayer* sqrt_layer = params->converter->network()->addUnary( in ConvertRsqrt() local3223 TFTRT_RETURN_ERROR_IF_NULLPTR(sqrt_layer, node_def.name()); in ConvertRsqrt()3226 *sqrt_layer->getOutput(0), nvinfer1::UnaryOperation::kRECIP); in ConvertRsqrt()