Home
last modified time | relevance | path

Searched defs:ConversionParams (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dtrt_optimization_pass.h43 struct ConversionParams { struct
44 string trt_logger_name = "DefaultLogger";
45 size_t max_batch_size = -1;
46 size_t max_workspace_size_bytes = 1 << 30;
47 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32;
48 int minimum_segment_size = 3;
50 bool is_dynamic_op = false;
52 int max_cached_engines = 1;
53 bool use_calibration = true;
77 ConversionParams params_; argument