OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add aten_op to caffe2 onnx (python) backend

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10579

Reviewed By: houseroad

Differential Revision: D9357837

fbshipit-source-id: 2cc6fedbaf088df7e11b52a91dfe3b8f0d7fd599
This commit is contained in:
Junjie Bai 2018-08-16 00:24:53 -07:00 committed by Facebook Github Bot
parent 76da0b34c2
commit 84427d26db
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
caffe2/python/onnx/backend.py
View File

@ -24,7 +24,7 @@ import itertools
import onnx.backend
import caffe2
from caffe2.python import core, workspace, rnn_cell, gru_cell
from caffe2.python import core, workspace, rnn_cell, gru_cell, dyndep
from caffe2.python.model_helper import ModelHelper
from caffe2.proto import caffe2_pb2
import caffe2.python.utils
@ -46,6 +46,9 @@ import caffe2.python._import_c_extension as C
import warnings
dyndep.InitOpsLibrary("@/caffe2/caffe2/contrib/aten:aten_op")
def force_unicode(s):
try:
return s.decode('utf-8')