Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_test.py68 trt_test.IsQuantizationMode(run_params.precision_mode) and
93 trt_test.IsQuantizationMode(run_params.precision_mode) and
102 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
106 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
121 run_params.precision_mode), "test non-INT8"
131 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
135 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
Ddynamic_input_shapes_test.py94 run_params.precision_mode)), "test dynamic engine and non-INT8"
98 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
102 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
Dtf_trt_integration_test_base.py104 return IsQuantizationMode(params.precision_mode) and params.use_calibration
281 precision_mode=run_params.precision_mode,
304 run_params.precision_mode)), "test either calibration or non-INT8"
316 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-02
320 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-02
467 precision_mode=conversion_params.precision_mode,
480 assert conversion_params.precision_mode == "INT8"
835 self._ToBytes(run_params.precision_mode),
1101 run_params.convert_online, run_params.precision_mode,
1111 for (precision_mode, convert_online, dynamic_engine, use_calibration) in opts:
[all …]
Dconst_broadcast_test.py57 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
61 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
Dreshape_transpose_test.py75 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
104 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
Dcast_test.py50 if run_params.precision_mode == "FP16":
Dmemory_alignment_test.py64 return 1.e-06 if run_params.precision_mode == "FP32" else 1.e-02
Dlru_cache_test.py66 run_params.precision_mode)), "test dynamic engine and non-INT8"
Dbase_test.py163 (run_params.precision_mode != "FP16") and
164 not (trt_test.IsQuantizationMode(run_params.precision_mode) and
Dcombined_nms_test.py96 not trt_test.IsQuantizationMode(run_params.precision_mode)
Dquantization_mnist_test.py150 precision_mode='INT8',
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h97 precision_mode(TrtPrecisionMode::FP32), in EngineInfo()
115 TrtPrecisionMode precision_mode; member
148 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size,
364 TrtPrecisionMode precision_mode, bool use_calibration,
379 const TrtPrecisionMode precision_mode; member
393 TrtPrecisionMode precision_mode, bool use_calibration,
457 TrtPrecisionMode precision_mode, bool use_calibration,
494 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode() function
610 Converter(TrtPrecisionMode precision_mode, bool use_calibration,
Dconvert_graph.cc425 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
454 calibrate_int8 ? TrtPrecisionMode::FP32 : info.precision_mode, in CreateTRTNode()
466 TF_RETURN_IF_ERROR(TrtPrecisionModeToName(info.precision_mode, &prec_string)); in CreateTRTNode()
684 if (params.precision_mode != TrtPrecisionMode::INT8 && in ConvertAfterShapes()
706 if (params.precision_mode == TrtPrecisionMode::FP16) { in ConvertAfterShapes()
742 TrtNodeValidator validator(static_graph_properties, params.precision_mode, in ConvertAfterShapes()
777 curr_engine.precision_mode = params.precision_mode; in ConvertAfterShapes()
Dconvert_graph.h40 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
Dtrt_optimization_pass.cc238 cp.precision_mode = precision_mode_; in Optimize()
Dconvert_nodes.cc1104 TrtPrecisionMode precision_mode, bool use_calibration, in OpConverterParams() argument
1111 precision_mode(precision_mode), in OpConverterParams()
1125 precision_mode(converter->precision_mode()), in OpConverterParams()
1138 TrtPrecisionMode precision_mode, bool use_calibration, in TrtNodeValidator() argument
1141 precision_mode_(precision_mode), in TrtNodeValidator()
1243 TrtPrecisionMode precision_mode, bool use_calibration, in Create() argument
1247 new Converter(precision_mode, use_calibration, trt_logger, in Create()
1253 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration, in Converter() argument
1257 : precision_mode_(precision_mode), in Converter()
1749 if (converter->precision_mode() == TrtPrecisionMode::INT8 && in PrepareTensorForShape()
[all …]
Dconvert_nodes_test.cc706 for (const TrtPrecisionMode precision_mode : in TEST()
708 TrtNodeValidator validator(graph_properties, precision_mode, in TEST()
723 if (precision_mode == TrtPrecisionMode::INT8) { in TEST()
1622 TrtNodeValidator validator(graph_properties, converter_->precision_mode(), in RunValidation()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py157 precision_mode=TrtPrecisionMode.FP32, argument
163 cls).__new__(cls, max_workspace_size_bytes, precision_mode,
185 if conversion_params.precision_mode not in supported_precision_modes:
188 "It should be one of {}").format(conversion_params.precision_mode,
293 conversion_params.precision_mode)
379 precision_mode=TrtPrecisionMode.FP32, argument
452 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
473 precision_mode=precision_mode,
960 conversion_params.precision_mode == TrtPrecisionMode.INT8 and
1241 precision_mode=TrtPrecisionMode.FP32, argument
[all …]
Dtrt_convert_windows.py84 precision_mode=TrtPrecisionMode.FP32, argument
Dtrt_convert_test.py206 precision_mode=(trt_convert.TrtPrecisionMode.INT8 if need_calibration
354 precision_mode=trt_convert.TrtPrecisionMode.FP32)
382 precision_mode=trt_convert.TrtPrecisionMode.FP32, argument
389 precision_mode=precision_mode,
507 precision_mode=trt_convert.TrtPrecisionMode.INT8,
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Drun_models.py151 for precision_mode in [
156 precision_mode=precision_mode,
157 use_calibration=(precision_mode == trt.TrtPrecisionMode.INT8))
Dmodel_handler.py460 precision_mode=trt_convert_params.precision_mode,
476 if (self.trt_convert_params.precision_mode == trt.TrtPrecisionMode.INT8 and
525 if (self.trt_convert_params.precision_mode == trt.TrtPrecisionMode.INT8 and
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.tensorrt.-conversion-params.pbtxt23 name: "precision_mode"
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py783 precision_mode=args.precision_mode,
801 precision_mode=args.precision_mode,
/external/tensorflow/
DRELEASE.md2215 * The `precision_mode` argument to `TrtGraphConverter` is now case