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

fix reshape test bug See merge request !580

Merge branch 'fix-reshape-test-bug' into 'master'
fix reshape test bug See merge request !580
049eb7d4 · 刘托 · 838a353e · dfb0f010 · 049eb7d4
显示空白变更内容
内联并排

Showing with 4 addition and 3 deletion

mace/ops/reshape_test.cc mace/ops/reshape_test.cc +4 -3

未找到文件。
--- a/mace/ops/reshape_test.cc
+++ b/mace/ops/reshape_test.cc
@@ -36,8 +36,9 @@ 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()},
+  net.AddInputFromArray<DeviceType::CPU, int32_t>(
+      "Shape",
+      {static_cast<index_t>(output_shape.size())},
      output_shape);

  // Run