Searched refs:recip_layer (Results 1 – 1 of 1) sorted by relevance
3225 nvinfer1::IUnaryLayer* recip_layer = params->converter->network()->addUnary( in ConvertRsqrt() local3227 TFTRT_RETURN_ERROR_IF_NULLPTR(recip_layer, node_def.name()); in ConvertRsqrt()3228 params->outputs->push_back(TRT_TensorOrWeights(recip_layer->getOutput(0))); in ConvertRsqrt()