mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
Summary: Addressed Dima's feedback. The proposal is here: https://fb.quip.com/TbQmAuqIznCf Pull Request resolved: https://github.com/pytorch/pytorch/pull/12384 Reviewed By: dzhulgakov Differential Revision: D10246743 Pulled By: houseroad fbshipit-source-id: c80db0c35d60ca32965275da705f2b1dfb2a7265
17 lines
421 B
Python
17 lines
421 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from caffe2.python import convert, workspace
|
|
from caffe2.proto import caffe2_pb2, torch_pb2
|
|
import unittest
|
|
import numpy as np
|
|
|
|
class TestOperator(unittest.TestCase):
|
|
def setUp(self):
|
|
workspace.ResetWorkspace()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|