def test_onnx_conv_ns_ap_s2(self):
dcr = dc.conv(self.onnx_dc_X2, self.onnx_dc_W, self.dc_B, "NOTSET",
dc.vectorInt([]), 1, dc.vectorInt([]),
dc.vectorInt([1, 0, 1, 0]), dc.vectorInt([2, 2]))
np.testing.assert_allclose(self.onnx_npr_ap_s2.astype(np.float32),
np.array(dcr.data()).astype(np.float32),
rtol=1e-3,
atol=1e-3)
def test_onnx_conv_valid_s2(self):
dcr = dc.conv(self.onnx_dc_X2, self.onnx_dc_W, self.dc_B, "VALID",
dc.vectorInt([]), 1, dc.vectorInt([]), dc.vectorInt([]),
dc.vectorInt([2, 2]))
np.testing.assert_allclose(self.onnx_npr_vl_s2.astype(np.float32),
np.array(dcr.data()).astype(np.float32),
rtol=1e-3,
atol=1e-3)
def test_onnx_conv_same_lower(self):
dcr = dc.conv(self.onnx_dc_X, self.onnx_dc_W, self.dc_B, "SAME_LOWER")
np.testing.assert_allclose(self.onnx_npr_su.astype(np.float32),
np.array(dcr.data()).astype(np.float32),
rtol=1e-3,
atol=1e-3)
def test_onnx_conv_valid(self):
dcr = dc.conv(self.onnx_dc_X, self.onnx_dc_W, self.dc_B, "VALID")
np.testing.assert_allclose(self.onnx_npr_vl.astype(np.float32),
np.array(dcr.data()).astype(np.float32),
rtol=1e-3,
atol=1e-3)
def test_onnx_conv_same_upper(self):
try:
dcr = dc.conv(self.onnx_dc_X, self.onnx_dc_BIGW, self.dc_B,
"VALID")
except ValueError as e:
assert e