Searched refs:weight_dims (Results 1 – 1 of 1) sorted by relevance
3065 nvinfer1::Dims weight_dims; in TfTensorToTrtWeights() local3066 GetTensorDimsWithProtoShape(tensor, &weight_dims); in TfTensorToTrtWeights()3067 *weights = weight_store->GetTempWeights(converted_dtype, weight_dims); in TfTensorToTrtWeights()