From f56c62cdcd32af676770d93925c674a37667e8e4 Mon Sep 17 00:00:00 2001 From: DEKHTIARJonathan Date: Tue, 21 Apr 2020 10:53:23 -0700 Subject: [PATCH] Env Var name change --- tensorflow/compiler/tf2tensorrt/segment/segment.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/compiler/tf2tensorrt/segment/segment.cc b/tensorflow/compiler/tf2tensorrt/segment/segment.cc index d65dc2cd0be..f46fde009c9 100644 --- a/tensorflow/compiler/tf2tensorrt/segment/segment.cc +++ b/tensorflow/compiler/tf2tensorrt/segment/segment.cc @@ -429,7 +429,7 @@ Status SegmentGraph(const Graph* tf_graph, // Getting the nodes blacklisted for conversion string tftrt_node_blacklist_str; TF_CHECK_OK(ReadStringFromEnvVar( - "TF_TRT_BLACKLIST_OPS", "", &tftrt_node_blacklist_str + "TF_TRT_OP_BLACKLIST", "", &tftrt_node_blacklist_str )); auto tftrt_node_blacklist = gtl::FlatSet{}; @@ -465,7 +465,7 @@ Status SegmentGraph(const Graph* tf_graph, LOG(WARNING) << "Blacklisted as TF-TRT candidate, " << "(Op type: " << node->tf_node()->type_string() << "), " << "(Op name: " << node->name() << "), " - << "(Reason: Blacklisted with the env var TF_TRT_BLACKLIST_OPS)"; + << "(Reason: Blacklisted with the env var TF_TRT_OP_BLACKLIST)"; unsupported_ops.emplace(node->tf_node()->type_string()); num_unsupported_ops++; node = nullptr;