提交 cb691d94 编写于 作者: 赵奇可

add truncate param to random input

上级 9b97813f
......@@ -133,7 +133,8 @@ void TestRandomResizeBicubic() {
OpsTestNet net;
// Add input data
net.AddRandomInput<D, float>("Input",
{batch, in_height, in_width, channels});
{batch, in_height, in_width, channels},
true, true);
net.TransformDataFormat<DeviceType::CPU, float>("Input", NHWC, "InputNCHW",
NCHW);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册