diff options
| author |  Shashi Shekhar <shashishekhar@google.com> | 2018-04-30 13:50:17 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-30 13:52:42 -0700 |
| commit | 57b7c7befa52ee4a205536c0552422a750cbcd21 (patch) | |
| tree | f98445fca00bba9956fa86696181b5e308c44772 /tensorflow/contrib/lite/interpreter.h | |
| parent | 1986f009218a5aa1653f91ed1f40e6321a91c922 (diff) | |
Fix a bug in profiler.
PiperOrigin-RevId: 194838948
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.h')
| -rw-r--r-- | tensorflow/contrib/lite/interpreter.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/interpreter.h b/tensorflow/contrib/lite/interpreter.h index 6f3433abcf..1074f64263 100644 --- a/tensorflow/contrib/lite/interpreter.h +++ b/tensorflow/contrib/lite/interpreter.h @@ -325,9 +325,7 @@ class Interpreter { void SetProfiler(profiling::Profiler* profiler) { profiler_ = profiler; } - profiling::Profiler* GetProfiler(profiling::Profiler* profiler) { - return profiler_; - } + profiling::Profiler* GetProfiler() { return profiler_; } // The default capacity of `tensors_` vector. static constexpr int kTensorsReservedCapacity = 128; |