diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-09-27 12:51:52 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 12:55:44 -0700 |
commit | 1084594657a5d139102ac794f84d1427a710e39a (patch) | |
tree | 108ac1628f966a213cf63b9f24e0f8b4add20d1e /tensorflow/contrib/lite/python | |
parent | 750466c6e6624d279de7f9a43accd682d487509c (diff) |
TFLite: Rename ResetVariableTensorsToZero -> ResetVariableTensors
PiperOrigin-RevId: 214820383
Diffstat (limited to 'tensorflow/contrib/lite/python')
3 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/contrib/lite/python/interpreter.py b/tensorflow/contrib/lite/python/interpreter.py index 1be61fe053..5700bf7892 100644 --- a/tensorflow/contrib/lite/python/interpreter.py +++ b/tensorflow/contrib/lite/python/interpreter.py @@ -253,5 +253,5 @@ class Interpreter(object): self._ensure_safe() self._interpreter.Invoke() - def reset_all_variables_to_zero(self): - return self._interpreter.ResetVariableTensorsToZero() + def reset_all_variables(self): + return self._interpreter.ResetVariableTensors() diff --git a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.cc b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.cc index 9ab05f3068..418f19a179 100644 --- a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.cc +++ b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.cc @@ -466,9 +466,9 @@ InterpreterWrapper* InterpreterWrapper::CreateWrapperCPPFromBuffer( error_msg); } -PyObject* InterpreterWrapper::ResetVariableTensorsToZero() { +PyObject* InterpreterWrapper::ResetVariableTensors() { TFLITE_PY_ENSURE_VALID_INTERPRETER(); - TFLITE_PY_CHECK(interpreter_->ResetVariableTensorsToZero()); + TFLITE_PY_CHECK(interpreter_->ResetVariableTensors()); Py_RETURN_NONE; } diff --git a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h index 641dd93db5..f5ca81e62a 100644 --- a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h +++ b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h @@ -65,7 +65,7 @@ class InterpreterWrapper { PyObject* TensorQuantization(int i) const; PyObject* SetTensor(int i, PyObject* value); PyObject* GetTensor(int i) const; - PyObject* ResetVariableTensorsToZero(); + PyObject* ResetVariableTensors(); // Returns a reference to tensor index i as a numpy array. The base_object // should be the interpreter object providing the memory. |