Home
last modified time | relevance | path

Searched refs:prec_string (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc137 string prec_string; in GetTrtRewriterConfig() local
139 TrtPrecisionModeToName(params.precision_mode, &prec_string)); in GetTrtRewriterConfig()
140 (*trt_parameter_map)["precision_mode"].set_s(prec_string); in GetTrtRewriterConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph.cc471 string prec_string; in CreateTRTNode() local
472 TF_RETURN_IF_ERROR(TrtPrecisionModeToName(info.precision_mode, &prec_string)); in CreateTRTNode()
501 .Attr("precision_mode", prec_string) in CreateTRTNode()