add workaround for scalar

src/onnx/onnx.cpp

@@ -137,24 +137,11 @@ struct onnx_parser
                 const std::vector<std::size_t>* s0 = &args[0]->get_shape().lens();
                 const std::vector<std::size_t>* s1 = &args[1]->get_shape().lens();
 
-                bool swapped = false;
                 // Make sure s0 is the smaller size
                 if(s0->size() > s1->size())
-                {
                     std::swap(s0, s1);
-                    swapped = true;
-                }
 
                 std::vector<std::size_t> output_lens(s1->size());
-                // if (s0->size() == 0)
-                // {
-                //     shape s = swapped ? args[0]->get_shape() : args[1]->get_shape();
-                //     auto l0 = prog.add_instruction(migraphx::op::scalar{s}, 1.0f);
-                //     return prog.add_instruction(x, l0, args[1]);
-                // }
-                // else
-                // {
-                    // Copy the larger vector to output_lens
                 auto offset = s1->size() - s0->size();
                 std::transform(s0->begin(),
                                s0->end(),
@@ -165,8 +152,6 @@ struct onnx_parser
                 auto l0 = prog.add_instruction(op::multibroadcast{output_lens}, args[0]);
                 auto l1 = prog.add_instruction(op::multibroadcast{output_lens}, args[1]);
                 return prog.add_instruction(x, l0, l1);
-                // }
-                
             }
             else
             {
@@ -602,6 +587,11 @@ struct onnx_parser
     static literal parse_tensor(const onnx::TensorProto& t)
     {
         std::vector<std::size_t> dims(t.dims().begin(), t.dims().end());
+        // in case of scalar constants in onnx file, use dims=1 to fill initializer data
+        if(dims.size() == 0)
+        {
+            dims = {1};
+        }
         if(t.has_raw_data())
         {
             const std::string& s = t.raw_data();