Searched refs:trt_logger (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/common/ |
| D | utils.cc | 57 void InitializeTrtPlugins(nvinfer1::ILogger* trt_logger) { in InitializeTrtPlugins() argument
63 bool plugin_initialized = initLibNvInferPlugins(trt_logger, ""); in InitializeTrtPlugins()
92 void MaybeInitializeTrtPlugins(nvinfer1::ILogger* trt_logger) { in MaybeInitializeTrtPlugins() argument
94 absl::call_once(once, InitializeTrtPlugins, trt_logger); in MaybeInitializeTrtPlugins()
|
| D | utils.h | 44 void MaybeInitializeTrtPlugins(nvinfer1::ILogger* trt_logger);
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 458 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch,
611 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch,
614 Status Init(nvinfer1::ILogger* trt_logger);
|
| D | convert_graph.cc | 448 auto trt_logger = GetLoggerRegistry()->LookUp(params.trt_logger_name); in CreateTRTNode() local
455 max_batch_size, info.max_workspace_size_bytes, input_shapes, trt_logger, in CreateTRTNode()
|
| D | convert_nodes.cc | 1244 nvinfer1::ILogger* trt_logger, const bool use_implicit_batch, in Create() argument
1247 new Converter(precision_mode, use_calibration, trt_logger, in Create()
1249 TF_RETURN_IF_ERROR(converter->Init(trt_logger)); in Create()
1254 nvinfer1::ILogger* trt_logger, in Converter() argument
1261 MaybeInitializeTrtPlugins(trt_logger); in Converter()
1265 Status Converter::Init(nvinfer1::ILogger* trt_logger) { in Init() argument
1267 trt_builder_.reset(nvinfer1::createInferBuilder(*trt_logger)); in Init()
6323 nvinfer1::ILogger* trt_logger, nvinfer1::IGpuAllocator* allocator, in ConvertGraphDefToEngine() argument
6332 auto statusor = Converter::Create(precision_mode, use_calibration, trt_logger, in ConvertGraphDefToEngine()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | BUILD | 254 srcs = ["utils/trt_logger.cc"],
255 hdrs = ["utils/trt_logger.h"],
|