Searched refs:precision_mode_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.h | 38 precision_mode_(TrtPrecisionMode::FP32), in name_()
66 TrtPrecisionMode precision_mode_; variable
|
| D | trt_optimization_pass.cc | 70 Uppercase(params.at("precision_mode").s()), &precision_mode_)); in Init()
232 if (use_calibration_ && precision_mode_ != TrtPrecisionMode::INT8) { in Optimize()
260 cp.precision_mode = precision_mode_; in Optimize()
|
| D | convert_graph.h | 55 const TrtPrecisionMode precision_mode_; variable
|
| D | convert_nodes.h | 464 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode()
571 const TrtPrecisionMode precision_mode_; variable
|
| D | convert_graph.cc | 71 : graph_properties_(graph_properties), precision_mode_(precision_mode) {} in TrtCandidateSelector()
83 precision_mode_, graph_properties_); in IsTensorRTCandidate()
|
| D | convert_nodes.cc | 996 precision_mode_(precision_mode), in Converter()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 116 TrtPrecisionMode precision_mode_; member in tensorflow::tensorrt::TRTEngineOp
217 TrtPrecisionModeFromName(precision_string, &precision_mode_)); in TRTEngineOp()
221 (use_calibration_ && precision_mode_ == TrtPrecisionMode::INT8 && in TRTEngineOp()
624 segment_graph_, precision_mode_, batch_size, workspace_size_, in GetEngine()
|