Searched defs:SetReshape (Results 1 – 1 of 1) sorted by relevance
678 ShuffleBuilder& SetReshape(const nvinfer1::Dims& dims) { in SetReshape() function683 ShuffleBuilder& SetReshape(nvinfer1::ITensor* shape) { in SetReshape() function