diff options
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert/utils.h')
-rw-r--r-- | tensorflow/contrib/tensorrt/convert/utils.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/utils.h b/tensorflow/contrib/tensorrt/convert/utils.h index f601c06701..8b5f4d614a 100644 --- a/tensorflow/contrib/tensorrt/convert/utils.h +++ b/tensorflow/contrib/tensorrt/convert/utils.h @@ -31,6 +31,8 @@ struct TrtDestroyer { template <typename T> using TrtUniquePtrType = std::unique_ptr<T, TrtDestroyer<T>>; +bool IsGoogleTensorRTEnabled(); + } // namespace tensorrt } // namespace tensorflow |