Searched refs:trt_tensor (Results 1 – 1 of 1) sorted by relevance
478 nvinfer1::ITensor* trt_tensor = layer->getOutput(0); in CreateConstantLayer() local484 trt_tensor->setType(trt_weights.type); in CreateConstantLayer()486 return trt_tensor; in CreateConstantLayer()4496 nvinfer1::ITensor* trt_tensor = layer->getOutput(0); in ConvertBinary() local4501 *trt_tensor, nvinfer1::UnaryOperation::kFLOOR); in ConvertBinary()4504 trt_tensor = layer->getOutput(0); in ConvertBinary()4507 params->outputs->push_back(TRT_TensorOrWeights(trt_tensor)); in ConvertBinary()