Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_int8_calibrator.cc41 engine_name_(engine_name) {} in TRTInt8Calibrator()
64 LOG(FATAL) << "FATAL " << engine_name_ << " input name '" << it.first in setBatch()
74 LOG(FATAL) << "cudaMemcpy " << engine_name_ << " for '" << it.first in setBatch()
139 VLOG(1) << "Got calibration data for " << engine_name_ << " @" << ptr in writeCalibrationCache()
143 VLOG(1) << "Destroying calibrator for " << engine_name_; in ~TRTInt8Calibrator()
Dtrt_int8_calibrator.h96 string engine_name_; member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h684 absl::string_view engine_name_; variable
Dconvert_nodes.cc1260 engine_name_(engine_name) { in Converter()
1676 SetLayerNameHelper(layer, engine_name_, node_def.name()); in SetLayerName()
1678 SetLayerNameHelper(layer, engine_name_, in SetLayerName()
1692 SetLayerNameHelper(layer, engine_name_, in SetLayerName()