Searched refs:input_dims_min (Results 1 – 1 of 1) sorted by relevance
178 nvinfer1::Dims input_dims_min = ConvertCudaDims(in_tensor.Shape()); in SetTensorRTNetworkInput() local179 input_dims_min.d[input_batchsize_index_] = 1; in SetTensorRTNetworkInput()181 input_dims_min.d[input_hw_index_] = 1; in SetTensorRTNetworkInput()182 input_dims_min.d[input_hw_index_ + 1] = 1; in SetTensorRTNetworkInput()184 …le_->setDimensions(in_tensor.Name().c_str(), nvinfer1::OptProfileSelector::kMIN, input_dims_min)) { in SetTensorRTNetworkInput()