Searched refs:TensorShapeToTrtDims (Results 1 – 1 of 1) sorted by relevance
202 inline nvinfer1::Dims TensorShapeToTrtDims(const TensorShapeType& shape, in TensorShapeToTrtDims() function218 *out = TensorShapeToTrtDims(tensor_shape, ignore_first_dim); in TensorShapeArrayToTrtDims()282 *trt_dims = TensorShapeToTrtDims(shape, /*ignore_first_dim=*/true); in ValidateTensorProperties()