experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #40158 from gaurav1086:fix_segmentation_fault_in_segment_graph

Browse Source 
PiperOrigin-RevId: 320132418
Change-Id: Ia355393a792bdd0d60bf4d179c7b43c8f6bdcfa9
This commit is contained in:
TensorFlower Gardener 2020-07-07 23:24:46 -07:00
commit faab6925c9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/compiler/tf2tensorrt/segment/segment.cc
View File

@ -721,6 +721,10 @@ Status SegmentGraph(const Graph* tf_graph,
std::vector<UnionFind<SimpleNode*>> node_segments;
for (int i = 0; i < graph->num_node_ids(); ++i) {
SimpleNode* node = graph->FindNodeId(i);
if (!node) {
VLOG(3) << "Node " << i << " doesn't exist in the graph";
continue;
}
auto exclude_node = [&](absl::string_view reason) {
VLOG(1) << "Not a TF-TRT candidate, "
<< "(Op type: " << node->tf_node()->type_string() << "), "