Home
last modified time | relevance | path

Searched refs:input_dims_max (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_subgraph.cc193 nvinfer1::Dims input_dims_max = ConvertCudaDims(in_tensor.Shape()); in SetTensorRTNetworkInput() local
195 …le_->setDimensions(in_tensor.Name().c_str(), nvinfer1::OptProfileSelector::kMAX, input_dims_max)) { in SetTensorRTNetworkInput()