diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-19 10:02:11 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-19 10:05:35 -0700 |
commit | 7f449920f8910561a4e57cc35b96fb7faf08ef98 (patch) | |
tree | 036ee28b10d59da6bb38a1e59625a5620b94924f /tensorflow/contrib/lite/interpreter.h | |
parent | 5fc2bdd2d5f624a6bad9e83b992029e3799ab64e (diff) |
Refresh allocations in the presence of dynamic tensors
PiperOrigin-RevId: 201193941
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.h')
-rw-r--r-- | tensorflow/contrib/lite/interpreter.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/interpreter.h b/tensorflow/contrib/lite/interpreter.h index 37961cd1dc..436c1007af 100644 --- a/tensorflow/contrib/lite/interpreter.h +++ b/tensorflow/contrib/lite/interpreter.h @@ -589,6 +589,11 @@ class Interpreter { bool allow_buffer_handle_output_ = false; + // Tracking bit for whether a tensor was resized in the course of an op + // invocation. This is a useful hint to ensure that dynamic tensor outputs + // trigger downstream reallocation after op invocation. + bool tensor_resized_since_op_invoke_ = false; + // Profiler for this interpreter instance. profiling::Profiler* profiler_; }; |