Function nptype

python/src/utils.cpp:50–63

Converts a TRT datatype to its corresponding numpy dtype.

// Converts a TRT datatype to its corresponding numpy dtype.
py::dtype nptype(nvinfer1::DataType type)
{
    switch (type)
    {
    case nvinfer1::DataType::kFLOAT: return py::dtype("f4");
    case nvinfer1::DataType::kHALF: return py::dtype("f2");
    case nvinfer1::DataType::kINT8: return py::dtype("i1");
    case nvinfer1::DataType::kINT32: return py::dtype("i4");
    case nvinfer1::DataType::kBOOL: return py::dtype("b1");
    case nvinfer1::DataType::kUINT8: return py::dtype("u1");
    case nvinfer1::DataType::kFP8: return py::dtype("f1");
    }
    return py::dtype("unknown");
}

nvinfer1::DataType type(py::dtype const& type)
{