OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #50091 from tensorflow/mihaimaruseac-patch-2-1

Browse Source 
Fix usage of TF_LITE_ENSURE_MSG in constructor
This commit is contained in:
Mihai Maruseac 2021-06-04 14:06:46 -07:00 committed by GitHub
commit a1d31d9090
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/core/subgraph.cc
View File

@ -151,11 +151,7 @@ class SubgraphGuard {
SubgraphGuard(TfLiteContext* context, bool* is_subgraph_in_use) SubgraphGuard(TfLiteContext* context, bool* is_subgraph_in_use)
: is_subgraph_in_use_(is_subgraph_in_use) { : is_subgraph_in_use_(is_subgraph_in_use) {
if (*is_subgraph_in_use_) { if (*is_subgraph_in_use_) {
TF_LITE_ENSURE_MSG( context->ReportError(context, "Subgraph is already in use.");
context, false,
"Subgraph is already in use. Using an interpreter or a subgraph in "
"multiple threads is not supported. Recursion in the graph is not "
"supported.");
status_ = kTfLiteError; status_ = kTfLiteError;
} else { } else {
*is_subgraph_in_use_ = true; *is_subgraph_in_use_ = true;