Merge pull request #29769 from compnerd:patch-1

PiperOrigin-RevId: 254174047
2019-06-20 04:56:10 -07:00 · 2019-06-20 04:56:10 -07:00 · 3bbf556706
commit 3bbf556706
parent 420473f1fb f235cc38fa
1 changed files with 5 additions and 1 deletions
--- a/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.i
+++ b/tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.i
@ -29,6 +29,10 @@ limitations under the License.
  $1 = reinterpret_cast<TfLiteDelegate*>(PyLong_AsVoidPtr($input));
 }

+%typemap(out) TfLiteDelegate* {
+  $result = PyLong_FromVoidPtr($1)
+}
+

 %include "tensorflow/lite/python/interpreter_wrapper/interpreter_wrapper.h"