Searched refs:trt_tensor (Results 1 – 1 of 1) sorted by relevance
451 nvinfer1::ITensor* trt_tensor = layer->getOutput(0); in CreateConstantLayer() local457 trt_tensor->setType(trt_dtype); in CreateConstantLayer()459 return trt_tensor; in CreateConstantLayer()