Update onnx verification

src/onnx/verify_onnx.cpp

@@ -12,9 +12,12 @@ migraph::argument run_cpu(std::string file)
 {
     auto p = migraph::parse_onnx(file);
     p.compile(migraph::cpu::cpu_target{});
-    auto s      = p.get_parameter_shape("Input3");
-    auto input3 = migraph::generate_argument(s);
-    auto out    = p.eval({{"Input3", input3}});
+    migraph::program::parameter_map m;
+    for(auto&& x : p.get_parameter_shapes())
+    {
+        m[x.first] = migraph::generate_argument(x.second);
+    }
+    auto out = p.eval(m);
     std::cout << p << std::endl;
     return out;
 }
@@ -22,14 +25,14 @@ migraph::argument run_cpu(std::string file)
 migraph::argument run_gpu(std::string file)
 {
     auto p = migraph::parse_onnx(file);
-    p.compile(migraph::cpu::cpu_target{});
-    auto s      = p.get_parameter_shape("Input3");
-    auto input3 = migraph::gpu::to_gpu(migraph::generate_argument(s));
+    p.compile(migraph::gpu::target{});
 
-    auto output = migraph::gpu::to_gpu(migraph::generate_argument(p.get_parameter_shape("output")));
-    auto handle = migraph::gpu::make_obj<migraph::gpu::miopen_handle>(&miopenCreate);
-
-    auto out = p.eval({{"Input3", input3}, {"output", output}});
+    migraph::program::parameter_map m;
+    for(auto&& x : p.get_parameter_shapes())
+    {
+        m[x.first] = migraph::gpu::to_gpu(migraph::generate_argument(x.second));
+    }
+    auto out = migraph::gpu::from_gpu(p.eval(m));
     std::cout << p << std::endl;
     return migraph::gpu::from_gpu(out);
 }

src/targets/gpu/lowering.cpp

@@ -138,7 +138,7 @@ struct miopen_pooling
     shape compute_shape(std::vector<shape> inputs) const
     {
         check_shapes{inputs, *this}.has(2).standard();
-        return op.compute_shape({inputs.at(1)});
+        return op.compute_shape({inputs.at(0)});
     }
     argument compute(context& ctx, shape output_shape, std::vector<argument> args) const
     {