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

Changing TF_TRT_BLACKLIST_OP to TF_TRT_BLACKLIST_OPS

Browse Source
This commit is contained in:
DEKHTIARJonathan 2020-04-20 18:57:26 -07:00
parent 47431b6178
commit e97fdb5539
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -436,7 +436,7 @@ Status SegmentGraph(const Graph* tf_graph,
// Getting the nodes blacklisted for conversion // Getting the nodes blacklisted for conversion
string tftrt_node_blacklist_str; string tftrt_node_blacklist_str;
TF_CHECK_OK(ReadStringFromEnvVar( TF_CHECK_OK(ReadStringFromEnvVar(
"TF_TRT_BLACKLIST_OP", "", &tftrt_node_blacklist_str "TF_TRT_BLACKLIST_OPS", "", &tftrt_node_blacklist_str
)); ));
auto tftrt_node_blacklist = gtl::FlatSet<string>{}; auto tftrt_node_blacklist = gtl::FlatSet<string>{};
@ -472,7 +472,7 @@ Status SegmentGraph(const Graph* tf_graph,
LOG(WARNING) << "Blacklisted as TF-TRT candidate, " LOG(WARNING) << "Blacklisted as TF-TRT candidate, "
<< "(Op type: " << node->tf_node()->type_string() << "), " << "(Op type: " << node->tf_node()->type_string() << "), "
<< "(Op name: " << node->name() << "), " << "(Op name: " << node->name() << "), "
<< "(Reason: Blacklisted with the env var TF_TRT_BLACKLIST_OP)"; << "(Reason: Blacklisted with the env var TF_TRT_BLACKLIST_OPS)";
unsupported_ops.emplace(node->tf_node()->type_string()); unsupported_ops.emplace(node->tf_node()->type_string());
num_unsupported_ops++; num_unsupported_ops++;
node = nullptr; node = nullptr;