Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_test.py78 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
102 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
111 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
115 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
126 return not trt_test.IsQuantizationMode(run_params.precision_mode)
136 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
140 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-01
Dtf_trt_integration_test_base.py73 return IsQuantizationMode(params.precision_mode) and params.use_calibration
180 precision_mode=run_params.precision_mode,
192 return not (IsQuantizationMode(run_params.precision_mode) and
201 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-02
205 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-02
225 conversion_params.precision_mode,
307 precision_mode=conversion_params.precision_mode,
320 assert conversion_params.precision_mode == "INT8"
447 self._ToBytes(run_params.precision_mode),
457 if (IsQuantizationMode(run_params.precision_mode) and
[all …]
Ddynamic_input_shapes_test.py95 not trt_test.IsQuantizationMode(run_params.precision_mode))
99 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
103 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
Dconst_broadcast_test.py71 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
75 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
Dreshape_transpose_test.py88 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
147 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
Dmemory_alignment_test.py75 return 1.e-06 if run_params.precision_mode == "FP32" else 1.e-02
Dlru_cache_test.py74 not trt_test.IsQuantizationMode(run_params.precision_mode))
Dint32_test.py78 return not trt_test.IsQuantizationMode(run_params.precision_mode)
Dbase_test.py194 return ((run_params.precision_mode != "FP16") and
195 not (trt_test.IsQuantizationMode(run_params.precision_mode) and
Dquantization_mnist_test.py145 precision_mode='INT8',
/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py29 precision_mode=trt_convert.TrtPrecisionMode.FP32, argument
43 precision_mode=precision_mode,
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py466 precision_mode=TrtPrecisionMode.FP32, argument
548 optimizer.parameter_map["precision_mode"].s = _to_bytes(precision_mode)
566 precision_mode=TrtPrecisionMode.FP32, argument
673 if precision_mode not in supported_precision_modes:
676 precision_mode, supported_precision_modes))
686 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
703 self._precision_mode = precision_mode
714 precision_mode=self._precision_mode,
786 precision_mode=TrtPrecisionMode.FP32, argument
867 precision_mode=precision_mode,
Dtrt_convert_test.py68 precision_mode="INT8",
187 precision_mode=(trt_convert.TrtPrecisionMode.INT8 if need_calibration
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h39 TrtPrecisionMode precision_mode);
64 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
87 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32,
Dconvert_nodes.h101 precision_mode(TrtPrecisionMode::FP32), in EngineInfo()
118 TrtPrecisionMode precision_mode; member
150 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size,
385 const TrtPrecisionMode precision_mode,
436 TrtPrecisionMode precision_mode, bool use_calibration);
464 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode() function
Dconvert_graph.cc70 TrtPrecisionMode precision_mode) in TrtCandidateSelector() argument
71 : graph_properties_(graph_properties), precision_mode_(precision_mode) {} in TrtCandidateSelector()
104 GraphDef* new_graph_def, TrtPrecisionMode precision_mode, in ConvertGraphDefToTensorRT() argument
164 precision_mode, parameters["precision_mode"].mutable_s())); in ConvertGraphDefToTensorRT()
503 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
517 calibrate_int8 ? TrtPrecisionMode::FP32 : info.precision_mode, in CreateTRTNode()
534 TF_RETURN_IF_ERROR(TrtPrecisionModeToName(info.precision_mode, &prec_string)); in CreateTRTNode()
790 if (params.precision_mode == TrtPrecisionMode::INT8) { in ConvertAfterShapes()
818 params.precision_mode); in ConvertAfterShapes()
854 curr_engine.precision_mode = params.precision_mode; in ConvertAfterShapes()
Dconvert_graph_test.cc101 for (const TrtPrecisionMode precision_mode : in TEST()
103 TrtCandidateSelector selector(graph_properties, precision_mode); in TEST()
117 if (precision_mode == TrtPrecisionMode::INT8) { in TEST()
Dconvert_nodes.cc938 const TrtPrecisionMode precision_mode, in ValidateNode() argument
949 is_supported_op = (precision_mode == TrtPrecisionMode::INT8); in ValidateNode()
994 TrtPrecisionMode precision_mode, bool use_calibration) in Converter() argument
996 precision_mode_(precision_mode), in Converter()
1264 if (precision_mode() == TrtPrecisionMode::INT8 && !use_calibration()) { in PrepareTensorForShape()
1299 if (precision_mode() != TrtPrecisionMode::INT8) return; in MaybeApplyQuantizationRanges()
1670 if (params->converter->precision_mode() == TrtPrecisionMode::FP16) { in BinaryTensorOpWeight()
1713 if (params->converter->precision_mode() == TrtPrecisionMode::INT8 && in BinaryTensorOpWeight()
1840 if (params->converter->precision_mode() == TrtPrecisionMode::FP16) { in ConvertConv2DHelper()
2983 if (params->converter->precision_mode() == TrtPrecisionMode::FP16) { in ConvertBiasAdd()
[all …]
Dtrt_optimization_pass.cc260 cp.precision_mode = precision_mode_; in Optimize()
Dconvert_nodes_test.cc1117 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32) { in BuildAndRun() argument
1127 if (precision_mode == TrtPrecisionMode::FP16) { in BuildAndRun()
1129 } else if (precision_mode == TrtPrecisionMode::INT8) { in BuildAndRun()
/external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/
Dplugin_test.py88 precision_mode="FP32",
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py675 precision_mode=args.precision_mode,