| 629 | } |
| 630 | |
| 631 | inline uint32_t getElementSize(nvinfer1::DataType t) noexcept |
| 632 | { |
| 633 | switch (t) |
| 634 | { |
| 635 | case nvinfer1::DataType::kINT32: return 4; |
| 636 | case nvinfer1::DataType::kFLOAT: return 4; |
| 637 | case nvinfer1::DataType::kHALF: return 2; |
| 638 | case nvinfer1::DataType::kBOOL: |
| 639 | case nvinfer1::DataType::kUINT8: |
| 640 | case nvinfer1::DataType::kINT8: |
| 641 | case nvinfer1::DataType::kFP8: return 1; |
| 642 | } |
| 643 | return 0; |
| 644 | } |
| 645 | |
| 646 | inline int64_t volume(nvinfer1::Dims const& dims, int32_t start, int32_t stop) |
| 647 | { |
no outgoing calls
no test coverage detected