Functions | |
int64_t | volume (const nvinfer1::Dims &d) |
void | printMem (const char *where) |
unsigned int | elementSize (nvinfer1::DataType t) |
std::ostream & | operator<< (std::ostream &o, const nvinfer1::Dims &dims) |
std::ostream & | operator<< (std::ostream &o, nvinfer1::DataType dt) |
nvinfer1::DimsCHW | getCHW (const nvinfer1::Dims &d) |
nvinfer1::DimsCHW | getCHWWithExpansion (const nvinfer1::Dims &d, int filler) |
int | combineIndexDimensions (int batchSize, const nvinfer1::Dims &d) |
template<typename A , typename B > | |
A | divUp (A m, B n) |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |