Searched refs:segment_graph_def_ (Results 1 – 1 of 1) sorted by relevance
162 GraphDef segment_graph_def_; member in tensorflow::tensorrt::TRTEngineOp325 return FunctionDefToGraphDef(func_handle, lib, &segment_graph_def_); in ImportSegmentGraphDef()844 segment_graph_def_, precision_mode_, batch_size, workspace_size_, in BuildEngine()909 if (segment_graph_def_.node().empty()) { in GetEngine()1079 this->segment_graph_def_, TrtPrecisionMode::INT8, in AllocateCalibrationResources()