Esempio n. 1
0
 def test_compatibility(self):
     test_mxnet.check_compatibility(self.model,
                                    self.x,
                                    self.fn,
                                    opset_version=self.opset_version)
     onnx_chainer.export(self.model,
                         self.x,
                         opset_version=self.opset_version)
Esempio n. 2
0
 def test_compatibility(self):
     if MXNET_OPSET_VERSION[self.name] is not None:
         for mxnet_opset_version in MXNET_OPSET_VERSION[self.name]:
             test_mxnet.check_compatibility(
                 self.model, self.x, self.fn, opset_version=mxnet_opset_version)
     for opset_version in range(1, onnx.defs.onnx_opset_version() + 1):
         onnx_chainer.export(self.model, self.x,
                             opset_version=opset_version)
Esempio n. 3
0
 def test_compatibility(self):
     test_mxnet.check_compatibility(self.model,
                                    self.x,
                                    self.fn,
                                    opset_version=self.opset_version)
     for opset_version in range(1, onnx.defs.onnx_opset_version() + 1):
         onnx_chainer.export(self.model,
                             self.x,
                             opset_version=opset_version)
Esempio n. 4
0
 def test_compatibility(self):
     test_mxnet.check_compatibility(self.model, self.x, self.fn)
Esempio n. 5
0
 def test_compatibility(self):
     if MXNET_SUPPORT[self.name]:
         test_mxnet.check_compatibility(self.model, self.x, self.fn)
     else:
         onnx_chainer.export(self.model, self.x)
Esempio n. 6
0
 def test_compatibility(self):
     test_mxnet.check_compatibility(
         self.model, self.x, self.fn, opset_version=self.opset_version)
Esempio n. 7
0
 def test_backend(self):
     test_mxnet.check_compatibility(self.model, (self.x1, self.x2),
                                    self.fn,
                                    opset_version=self.opset_version)
     onnx_chainer.export(self.model, (self.x1, self.x2),
                         opset_version=self.opset_version)