Searched refs:tensor_or_weights (Results 1 – 3 of 3) sorted by relevance
414 TRT_TensorOrWeights* tensor_or_weights);
1149 TRT_TensorOrWeights* tensor_or_weights) { in ConvertToTensorOrWeights() argument1162 return ConvertConstToWeights(node_def, inputs, tensor_or_weights); in ConvertToTensorOrWeights()1184 *tensor_or_weights = TRT_TensorOrWeights(trt_dtype, trt_dims, batch_size); in ConvertToTensorOrWeights()1209 TRT_TensorOrWeights tensor_or_weights; in IsTensorRTCandidate() local1212 &tensor_or_weights); in IsTensorRTCandidate()1218 inputs.push_back(tensor_or_weights); in IsTensorRTCandidate()1363 TRT_TensorOrWeights tensor_or_weights; in RenameAndMarkOutputTensors() local1365 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()1366 if (!tensor_or_weights.is_tensor()) { in RenameAndMarkOutputTensors()1370 nvinfer1::ITensor* tensor = tensor_or_weights.tensor(); in RenameAndMarkOutputTensors()
536 TRT_TensorOrWeights* tensor_or_weights) { in ConvertToTensorOrWeights() argument546 tensor_or_weights); in ConvertToTensorOrWeights()