diff options
author | 2018-06-12 22:23:01 -0700 | |
---|---|---|
committer | 2018-06-12 22:23:01 -0700 | |
commit | 4979c54d90a7fdd7429feb50edd1520d819c9653 (patch) | |
tree | be90288f92e8ca25884cc26cef48d686156e9b7d /tensorflow/contrib/tensorrt/trt_conversion.i | |
parent | ee169363b5583ae7e16461aaf1588d6a0a9aa710 (diff) |
Further changes for review requests
Diffstat (limited to 'tensorflow/contrib/tensorrt/trt_conversion.i')
-rw-r--r-- | tensorflow/contrib/tensorrt/trt_conversion.i | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/tensorrt/trt_conversion.i b/tensorflow/contrib/tensorrt/trt_conversion.i index 80bb14accf..226454dbab 100644 --- a/tensorflow/contrib/tensorrt/trt_conversion.i +++ b/tensorflow/contrib/tensorrt/trt_conversion.i @@ -69,13 +69,13 @@ PyObject* version_helper(version_struct* in) { } /* Define converters for vector<int> */ template<> - bool _PyObjAs(PyObject *pyobj, int* dest) { +bool _PyObjAs(PyObject *pyobj, int* dest) { *dest = PyLong_AsLong(pyobj); return true; } template<> - PyObject *_PyObjFrom(const int& src) { +PyObject *_PyObjFrom(const int& src) { return PyLong_FromLong(src); } |