Home
last modified time | relevance | path

Searched refs:trt_type (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc625 nvinfer1::DataType trt_type(nvinfer1::DataType::kFLOAT); in GetTrtWeights() local
626 TF_CHECK_OK(ConvertDType(type_, &trt_type)); in GetTrtWeights()
627 return nvinfer1::Weights{trt_type, GetValues(), count()}; in GetTrtWeights()