Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc337 &info->segment_graph_def, &scope_name)); in GetEngineInfo()
516 info.segment_graph_def, in CreateTRTNode()
527 segment_string = info.segment_graph_def.SerializeAsString(); in CreateTRTNode()
530 segment_string = info.segment_graph_def.SerializeAsString(); in CreateTRTNode()
863 &graph, curr_engine.segment_graph_def, curr_engine.engine_name); in ConvertAfterShapes()
871 engine_bytes_size.push_back(curr_engine.segment_graph_def.ByteSizeLong()); in ConvertAfterShapes()
882 f << engine_segments.at(t).segment_graph_def.SerializeAsString(); in ConvertAfterShapes()
Dconvert_nodes.h106 GraphDef segment_graph_def; member