diff options
author | 2018-02-06 14:04:35 -0800 | |
---|---|---|
committer | 2018-02-06 14:04:35 -0800 | |
commit | bfe8b85cad3be1a82234500fce3064c98dd20d09 (patch) | |
tree | 6fc5259ddf3941d074b009253f1b2e3faf008fe2 /tensorflow/contrib/tensorrt/trt_conversion.i | |
parent | 360ae1cd2c7a73f96e8fb4fd601521dc108246cf (diff) |
1. Use tensorflow::string instead of std::string
2. Fix swig problem in trt_conversion.i
3. Fix c++ styles
4. Fix BUILD dependencies
Diffstat (limited to 'tensorflow/contrib/tensorrt/trt_conversion.i')
-rw-r--r-- | tensorflow/contrib/tensorrt/trt_conversion.i | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/contrib/tensorrt/trt_conversion.i b/tensorflow/contrib/tensorrt/trt_conversion.i index f085380054..1570ee6f04 100644 --- a/tensorflow/contrib/tensorrt/trt_conversion.i +++ b/tensorflow/contrib/tensorrt/trt_conversion.i @@ -13,15 +13,13 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ -/* wrap trt_conversion */ +/* Wrap trt_conversion */ %{ #define SWIG_FILE_WITH_INIT %} -%include "std_string.i" %include "std_pair.i" -%include "tensorflow/python/lib/core/strings.i" %include "tensorflow/python/platform/base.i" -%template(StringPair) std::pair<string,string>; +%template(StringPair) std::pair<string, string>; %template() std::pair<swig::SwigPtr_PyObject, swig::SwigPtr_PyObject>; %{ |