提交 049eb7d4 编写于 作者: 刘托

Merge branch 'fix-reshape-test-bug' into 'master'

fix reshape test bug

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