Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc24 Status TrtPrecisionModeToName(TrtPrecisionMode mode, string* name) { in TrtPrecisionModeToName()
26 case TrtPrecisionMode::FP32: in TrtPrecisionModeToName()
29 case TrtPrecisionMode::FP16: in TrtPrecisionModeToName()
32 case TrtPrecisionMode::INT8: in TrtPrecisionModeToName()
41 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode) { in TrtPrecisionModeFromName()
43 *mode = TrtPrecisionMode::FP32; in TrtPrecisionModeFromName()
45 *mode = TrtPrecisionMode::FP16; in TrtPrecisionModeFromName()
47 *mode = TrtPrecisionMode::INT8; in TrtPrecisionModeFromName()
Dconvert_graph.h39 TrtPrecisionMode precision_mode);
55 const TrtPrecisionMode precision_mode_;
64 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32;
87 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32,
Dutils.h36 enum class TrtPrecisionMode { FP32, FP16, INT8 }; enum
38 Status TrtPrecisionModeToName(TrtPrecisionMode mode, string* name);
40 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode);
Dconvert_nodes.h101 precision_mode(TrtPrecisionMode::FP32), in EngineInfo()
118 TrtPrecisionMode precision_mode;
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()
571 const TrtPrecisionMode precision_mode_;
Dconvert_graph_test.cc101 for (const TrtPrecisionMode precision_mode : in TEST()
102 {TrtPrecisionMode::FP32, TrtPrecisionMode::INT8}) { in TEST()
117 if (precision_mode == TrtPrecisionMode::INT8) { in TEST()
Dtrt_optimization_pass.h38 precision_mode_(TrtPrecisionMode::FP32), in name_()
66 TrtPrecisionMode precision_mode_;
Dconvert_nodes_test.cc494 ExpectStatus(validator_.ValidateNode(node_def, {}, TrtPrecisionMode::FP32, in TEST_F()
500 TF_EXPECT_OK(validator_.ValidateNode(node_def, {}, TrtPrecisionMode::FP32, in TEST_F()
506 ExpectStatus(validator_.ValidateNode(node_def, {}, TrtPrecisionMode::FP32, in TEST_F()
513 ExpectStatus(validator_.ValidateNode(node_def, {}, TrtPrecisionMode::FP32, in TEST_F()
524 converter_.reset(new Converter(network_.get(), TrtPrecisionMode::FP32, in ConverterTest()
843 Converter int8_converter(/*trt_network=*/nullptr, TrtPrecisionMode::INT8, in TEST_F()
1012 gdef, TrtPrecisionMode::FP32, /*max_batch_size=*/1, in RunConvertGraphDefToEngine()
1117 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32) { in BuildAndRun()
1127 if (precision_mode == TrtPrecisionMode::FP16) { in BuildAndRun()
1129 } else if (precision_mode == TrtPrecisionMode::INT8) { in BuildAndRun()
[all …]
Dconvert_graph.cc70 TrtPrecisionMode precision_mode) in TrtCandidateSelector()
104 GraphDef* new_graph_def, TrtPrecisionMode precision_mode, in ConvertGraphDefToTensorRT()
503 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
517 calibrate_int8 ? TrtPrecisionMode::FP32 : info.precision_mode, in CreateTRTNode()
790 if (params.precision_mode == TrtPrecisionMode::INT8) { in ConvertAfterShapes()
Dconvert_nodes.cc938 const TrtPrecisionMode precision_mode, in ValidateNode()
949 is_supported_op = (precision_mode == TrtPrecisionMode::INT8); in ValidateNode()
994 TrtPrecisionMode precision_mode, bool use_calibration) 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()
3210 if (params->converter->precision_mode() == TrtPrecisionMode::INT8 && in ConvertRsqrt()
[all …]
Dtrt_optimization_pass.cc232 if (use_calibration_ && precision_mode_ != TrtPrecisionMode::INT8) { in Optimize()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py440 class TrtPrecisionMode(object): class
447 return [TrtPrecisionMode.FP32, TrtPrecisionMode.FP16, TrtPrecisionMode.INT8]
466 precision_mode=TrtPrecisionMode.FP32,
566 precision_mode=TrtPrecisionMode.FP32,
672 supported_precision_modes = TrtPrecisionMode.supported_precision_modes()
686 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
786 precision_mode=TrtPrecisionMode.FP32,
Dtrt_convert_test.py187 precision_mode=(trt_convert.TrtPrecisionMode.INT8 if need_calibration
188 else trt_convert.TrtPrecisionMode.FP32),
/external/tensorflow/tensorflow/contrib/tensorrt/python/
Dtrt_convert.py29 precision_mode=trt_convert.TrtPrecisionMode.FP32,
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc116 TrtPrecisionMode precision_mode_;
221 (use_calibration_ && precision_mode_ == TrtPrecisionMode::INT8 && in TRTEngineOp()
712 *segment_graph, TrtPrecisionMode::INT8, in AllocateCalibrationResources()