Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
未验证 提交 de2166c0 编写于 作者: H houj04 提交者： GitHub
浏览文件
操作

[XPU] fix unit test of test_pad3d_op_xpu. (#51962)

上级 153351e1
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
python/paddle/fluid/tests/unittests/xpu/test_pad3d_op_xpu.py
浏览文件 @ de2166c0
......@@ -189,7 +189,7 @@ class XPUTestPad3dOp(XPUOpTestWrapper):
value = 100
input_data = np.random.rand(*input_shape).astype(self.dtype)
x = paddle.static.data(
name="x", shape=input_shape, dtype="float32"
name="x", shape=input_shape, dtype=self.dtype
)
result = F.pad(
x=x, pad=pad, value=value, mode=mode, data_format="NCDHW"
......@@ -212,7 +212,7 @@ class XPUTestPad3dOp(XPUOpTestWrapper):
mode = "reflect"
input_data = np.random.rand(*input_shape).astype(self.dtype)
x = paddle.static.data(
name="x", shape=input_shape, dtype="float32"
name="x", shape=input_shape, dtype=self.dtype
)
result1 = F.pad(x=x, pad=pad, mode=mode, data_format="NCDHW")
result2 = F.pad(x=x, pad=pad, mode=mode, data_format="NDHWC")
......@@ -240,7 +240,7 @@ class XPUTestPad3dOp(XPUOpTestWrapper):
mode = "replicate"
input_data = np.random.rand(*input_shape).astype(self.dtype)
x = paddle.static.data(
name="x", shape=input_shape, dtype="float32"
name="x", shape=input_shape, dtype=self.dtype
)
result1 = F.pad(x=x, pad=pad, mode=mode, data_format="NCDHW")
result2 = F.pad(x=x, pad=pad, mode=mode, data_format="NDHWC")
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或