Searched refs:sqrt_layer (Results 1 – 1 of 1) sorted by relevance
4539 nvinfer1::IUnaryLayer* sqrt_layer = params->converter->network()->addUnary( in ConvertRsqrt() local4541 TFTRT_RETURN_ERROR_IF_NULLPTR(sqrt_layer, node_def.name()); in ConvertRsqrt()4542 params->converter->SetLayerName(sqrt_layer, node_def, "sqrt"); in ConvertRsqrt()4545 *sqrt_layer->getOutput(0), nvinfer1::UnaryOperation::kRECIP); in ConvertRsqrt()