Searched defs:nvinfer1 (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/convert/
H A Dconvert_nodes.cc56 nvinfer1::DataType* trt_dtype) {
59 *trt_dtype = nvinfer1::DataType::kFLOAT;
62 *trt_dtype = nvinfer1::DataType::kINT8;
65 *trt_dtype = nvinfer1::DataType::kHALF;
73 inline nvinfer1::Dims GetTensorShape(const tensorflow::Tensor& tensor) {
74 nvinfer1::Dims dims;
82 inline int64_t GetShapeSize(nvinfer1::Dims shape) {
92 const nvinfer1::DimsHW& stride, const nvinfer1::DimsHW& kernel,
118 nvinfer1
[all...]

Completed in 126 milliseconds