Home
last modified time | relevance | path

Searched refs:TrtPrecisionModeFromName (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h40 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode);
Dutils.cc41 Status TrtPrecisionModeFromName(const string& name, TrtPrecisionMode* mode) { in TrtPrecisionModeFromName() function
Dtrt_optimization_pass.cc69 TF_RETURN_IF_ERROR(TrtPrecisionModeFromName( in Init()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc217 TrtPrecisionModeFromName(precision_string, &precision_mode_)); in TRTEngineOp()