Searched refs:GetTensorDimsWithProtoShape (Results 1 – 1 of 1) sorted by relevance
3027 void GetTensorDimsWithProtoShape(const Tensor& tensor, nvinfer1::Dims* dims) { in GetTensorDimsWithProtoShape() function3066 GetTensorDimsWithProtoShape(tensor, &weight_dims); in TfTensorToTrtWeights()