Added MNIST driver

src/include/rtg/tensor_view.hpp

@@ -29,12 +29,14 @@ struct tensor_view
     template <class... Ts, RTG_REQUIRES(std::is_integral<Ts>{}...)>
     const T& operator()(Ts... xs) const
     {
+        assert(m_shape.index({static_cast<std::size_t>(xs)...}) < m_shape.bytes() / sizeof(T));
         return m_data[m_shape.index({static_cast<std::size_t>(xs)...})];
     }
 
     template <class... Ts, RTG_REQUIRES(std::is_integral<Ts>{}...)>
     T& operator()(Ts... xs)
     {
+        assert(m_shape.index({static_cast<std::size_t>(xs)...}) < m_shape.bytes() / sizeof(T));
         return m_data[m_shape.index({static_cast<std::size_t>(xs)...})];
     }
 

src/onnx/mnist.cpp

@@ -136,7 +136,8 @@ int main(int argc, char const* argv[])
         std::string file = argv[1];
         auto prog        = rtg::parse_onnx(file);
         prog.compile(rtg::cpu::cpu_target{});
-        auto s = prog.get_parameter_shape("Input3");
+        // auto s = prog.get_parameter_shape("Input3");
+        auto s = rtg::shape{rtg::shape::float_type, {1, 1, 28, 28}};
         std::cout << s << std::endl;
         auto input3 = rtg::argument{s, input.data()};
         auto out    = prog.eval({{"Input3", input3}});