Searched defs:precision_mode (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensorrt/python/ |
| D | trt_convert.py | 29 precision_mode=trt_convert.TrtPrecisionMode.FP32, argument
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 466 precision_mode=TrtPrecisionMode.FP32, argument
566 precision_mode=TrtPrecisionMode.FP32, argument
786 precision_mode=TrtPrecisionMode.FP32, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.h | 64 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32; member
|
| D | convert_nodes.h | 118 TrtPrecisionMode precision_mode; member
464 TrtPrecisionMode precision_mode() const { return precision_mode_; } in precision_mode() function
|
| D | convert_graph.cc | 70 TrtPrecisionMode precision_mode) in TrtCandidateSelector()
104 GraphDef* new_graph_def, TrtPrecisionMode precision_mode, in ConvertGraphDefToTensorRT()
|
| D | convert_nodes.cc | 938 const TrtPrecisionMode precision_mode, in ValidateNode()
994 TrtPrecisionMode precision_mode, bool use_calibration) in Converter()
4056 const GraphDef& gdef, TrtPrecisionMode precision_mode, int max_batch_size, in ConvertGraphDefToEngine()
|
| D | convert_nodes_test.cc | 1117 TrtPrecisionMode precision_mode = TrtPrecisionMode::FP32) { in BuildAndRun()
|