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