| 60 | } |
| 61 | |
| 62 | nvinfer1::Dims toDims(std::vector<int32_t> const& vec) |
| 63 | { |
| 64 | int32_t limit = static_cast<int32_t>(nvinfer1::Dims::MAX_DIMS); |
| 65 | if (static_cast<int32_t>(vec.size()) > limit) |
| 66 | { |
| 67 | sample::gLogWarning << "Vector too long, only first 8 elements are used in dimension." << std::endl; |
| 68 | } |
| 69 | // Pick first nvinfer1::Dims::MAX_DIMS elements |
| 70 | nvinfer1::Dims dims{std::min(static_cast<int32_t>(vec.size()), limit), {}}; |
| 71 | std::copy_n(vec.begin(), dims.nbDims, std::begin(dims.d)); |
| 72 | return dims; |
| 73 | } |
| 74 | |
| 75 | void loadFromFile(std::string const& fileName, char* dst, size_t size) |
| 76 | { |
no test coverage detected