Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc25 Status TrtPrecisionModeToName(const TrtPrecisionMode mode, string* name) { in TrtPrecisionModeToName()
27 case TrtPrecisionMode::FP32: in TrtPrecisionModeToName()
30 case TrtPrecisionMode::FP16: in TrtPrecisionModeToName()
33 case TrtPrecisionMode::INT8: in TrtPrecisionModeToName()
43 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode) { in TrtPrecisionModeFromName()
45 *mode = TrtPrecisionMode::FP32; in TrtPrecisionModeFromName()
47 *mode = TrtPrecisionMode::FP16; in TrtPrecisionModeFromName()
49 *mode = TrtPrecisionMode::INT8; in TrtPrecisionModeFromName()
117 string DebugString(const TrtPrecisionMode mode) { in DebugString()
Dconvert_nodes.h97 precision_mode(TrtPrecisionMode::FP32), in EngineInfo()
115 TrtPrecisionMode precision_mode;
148 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size,
364 TrtPrecisionMode precision_mode, bool use_calibration,
379 const TrtPrecisionMode precision_mode;
393 TrtPrecisionMode precision_mode, bool use_calibration,
430 const TrtPrecisionMode precision_mode_;
457 TrtPrecisionMode precision_mode, bool use_calibration,
494 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode()
610 Converter(TrtPrecisionMode precision_mode, bool use_calibration,
[all …]
Dconvert_graph.h40 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32;
Dutils.h55 enum class TrtPrecisionMode { FP32, FP16, INT8 }; enum
57 Status TrtPrecisionModeToName(const TrtPrecisionMode mode, string* name);
59 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode);
106 string DebugString(const TrtPrecisionMode mode);
Dtrt_optimization_pass.h38 precision_mode_(TrtPrecisionMode::FP32), in name_()
70 TrtPrecisionMode precision_mode_;
Dconvert_nodes_test.cc542 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in ConvertToTensorOrWeights()
557 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in TEST_F()
628 TrtNodeValidator validator(graph_properties, TrtPrecisionMode::FP32, in TEST_F()
706 for (const TrtPrecisionMode precision_mode : in TEST()
707 {TrtPrecisionMode::FP32, TrtPrecisionMode::INT8}) { in TEST()
723 if (precision_mode == TrtPrecisionMode::INT8) { in TEST()
739 std::move(Converter::Create(TrtPrecisionMode::FP32, in Reset()
1096 auto int8_converter = Converter::Create(TrtPrecisionMode::INT8, in TEST_F()
1289 gdef, TrtPrecisionMode::FP32, /*max_batch_size=*/1, in RunConvertGraphDefToEngine()
1369 void Reset(TrtPrecisionMode precision_mode_to_test = TrtPrecisionMode::FP32, in Reset()
[all …]
Dconvert_graph.cc425 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
454 calibrate_int8 ? TrtPrecisionMode::FP32 : info.precision_mode, in CreateTRTNode()
684 if (params.precision_mode != TrtPrecisionMode::INT8 && in ConvertAfterShapes()
706 if (params.precision_mode == TrtPrecisionMode::FP16) { in ConvertAfterShapes()
Dtrt_optimization_pass.cc207 if (use_calibration_ && precision_mode_ != TrtPrecisionMode::INT8) { in Optimize()
Dconvert_nodes.cc1104 TrtPrecisionMode precision_mode, bool use_calibration, in OpConverterParams()
1138 TrtPrecisionMode precision_mode, bool use_calibration, in TrtNodeValidator()
1195 is_supported_op = (precision_mode_ == TrtPrecisionMode::INT8); in IsTensorRTCandidate()
1243 TrtPrecisionMode precision_mode, bool use_calibration, in Create()
1253 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration, in Converter()
1485 if (precision_mode_ == TrtPrecisionMode::FP16) { in BuildCudaEngine()
1487 } else if (precision_mode_ == TrtPrecisionMode::INT8) { in BuildCudaEngine()
1518 if (precision_mode_ == TrtPrecisionMode::FP16) { in BuildCudaEngine()
1520 } else if (precision_mode_ == TrtPrecisionMode::INT8) { in BuildCudaEngine()
1749 if (converter->precision_mode() == TrtPrecisionMode::INT8 && in PrepareTensorForShape()
[all …]
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_windows.py31 class TrtPrecisionMode(object): class
84 precision_mode=TrtPrecisionMode.FP32,
Dtrt_convert.py100 class TrtPrecisionMode(object): class
108 TrtPrecisionMode.FP32, TrtPrecisionMode.FP16, TrtPrecisionMode.INT8
157 precision_mode=TrtPrecisionMode.FP32,
184 supported_precision_modes = TrtPrecisionMode.supported_precision_modes()
379 precision_mode=TrtPrecisionMode.FP32,
452 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
960 conversion_params.precision_mode == TrtPrecisionMode.INT8 and
1241 precision_mode=TrtPrecisionMode.FP32,
Dtrt_convert_test.py206 precision_mode=(trt_convert.TrtPrecisionMode.INT8 if need_calibration
207 else trt_convert.TrtPrecisionMode.FP32),
354 precision_mode=trt_convert.TrtPrecisionMode.FP32)
382 precision_mode=trt_convert.TrtPrecisionMode.FP32,
507 precision_mode=trt_convert.TrtPrecisionMode.INT8,
/external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/
Drun_models.py152 trt.TrtPrecisionMode.FP32, trt.TrtPrecisionMode.FP16,
153 trt.TrtPrecisionMode.INT8
157 use_calibration=(precision_mode == trt.TrtPrecisionMode.INT8))
Dmodel_handler.py476 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/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc165 TrtPrecisionMode precision_mode_;
381 (use_calibration_ && precision_mode_ == TrtPrecisionMode::INT8 && in TRTEngineOp()
1079 this->segment_graph_def_, TrtPrecisionMode::INT8, in AllocateCalibrationResources()