提交 dfb0f010 编写于 作者: L liutuo

fix reshape test bug

上级 23b2a77e
...@@ -36,9 +36,10 @@ void TestReshape(const std::vector<index_t> &org_shape, ...@@ -36,9 +36,10 @@ void TestReshape(const std::vector<index_t> &org_shape,
// Add input data // Add input data
net.AddRandomInput<DeviceType::CPU, float>("Input", org_shape); net.AddRandomInput<DeviceType::CPU, float>("Input", org_shape);
net.AddInputFromArray<DeviceType::CPU, int32_t>("Shape", net.AddInputFromArray<DeviceType::CPU, int32_t>(
{output_shape.size()}, "Shape",
output_shape); {static_cast<index_t>(output_shape.size())},
output_shape);
// Run // Run
net.RunOp(); net.RunOp();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册