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'))
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')