Example #1
0
 def test_apply(self):
     oopb = _cmn.onnx_ops.OnnxOperatorBuilder(
         _cmn.OnnxObjectContainer(_proto.get_opset_number_from_onnx()),
         _cmn.InterimContext('_curr'))
     value = oopb.apply_add(
         (np.array([[1.0], [0.5]], dtype='float32'),
          ('_i1', oopb.float, np.array([2.0], dtype='float32'))), 'add')
     self.assertTrue(value[0].startswith('add'))
Example #2
0
def test_apply():
    oopb = _cmn.onnx_ops.OnnxOperatorBuilder(
        _cmn.OnnxObjectContainer(get_maximum_opset_supported()),
        _cmn.InterimContext('_curr'))
    value = oopb.apply_add(
        (np.array([[1.0], [0.5]], dtype='float32'),
         ('_i1', oopb.float, np.array([2.0], dtype='float32'))), 'add')
    assert value[0].startswith('add')