Searched refs:recip_layer (Results 1 – 1 of 1) sorted by relevance
4544 nvinfer1::IUnaryLayer* recip_layer = params->converter->network()->addUnary( in ConvertRsqrt() local4546 TFTRT_RETURN_ERROR_IF_NULLPTR(recip_layer, node_def.name()); in ConvertRsqrt()4547 params->converter->SetLayerName(recip_layer, node_def, "recip"); in ConvertRsqrt()4548 params->outputs->push_back(TRT_TensorOrWeights(recip_layer->getOutput(0))); in ConvertRsqrt()