Commit 84ea43b5 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

WIP

parent c405c3bc
...@@ -12,6 +12,7 @@ ...@@ -12,6 +12,7 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// ---------------------------------------------------------------------------- // ----------------------------------------------------------------------------
#include <cassert>
#include <cstdlib> #include <cstdlib>
#include <fstream> #include <fstream>
#include <memory> #include <memory>
...@@ -25,7 +26,6 @@ ...@@ -25,7 +26,6 @@
#include "ngraph/codegen/compiler.hpp" #include "ngraph/codegen/compiler.hpp"
#include "ngraph/codegen/execution_engine.hpp" #include "ngraph/codegen/execution_engine.hpp"
#include "ngraph/descriptor/input.hpp" #include "ngraph/descriptor/input.hpp"
#include "ngraph/descriptor/layout/dense_tensor_view_layout.hpp"
#include "ngraph/descriptor/output.hpp" #include "ngraph/descriptor/output.hpp"
#include "ngraph/descriptor/primary_tensor_view.hpp" #include "ngraph/descriptor/primary_tensor_view.hpp"
#include "ngraph/file_util.hpp" #include "ngraph/file_util.hpp"
......
...@@ -49,7 +49,7 @@ namespace ngraph ...@@ -49,7 +49,7 @@ namespace ngraph
throw ngraph_error("Axis is out of bounds"); throw ngraph_error("Axis is out of bounds");
} }
strides.emplace_back(shape[*it]); strides.emplace_back(s);
s *= shape[*it]; s *= shape[*it];
} }
std::reverse(strides.begin(), strides.end()); std::reverse(strides.begin(), strides.end());
......
...@@ -34,6 +34,7 @@ namespace ngraph ...@@ -34,6 +34,7 @@ namespace ngraph
public: public:
LayoutDescriptor(const ngraph::descriptor::TensorView& tv, LayoutDescriptor(const ngraph::descriptor::TensorView& tv,
const AxisVector& tv_axis_order); const AxisVector& tv_axis_order);
~LayoutDescriptor() { }
size_t get_size() override { return size; } size_t get_size() override { return size; }
size_t get_offset() const { return offset; } size_t get_offset() const { return offset; }
......
...@@ -13,6 +13,7 @@ ...@@ -13,6 +13,7 @@
// ---------------------------------------------------------------------------- // ----------------------------------------------------------------------------
#include <algorithm> #include <algorithm>
#include <memory>
#include "cpu_layout.hpp" #include "cpu_layout.hpp"
#include "ngraph/descriptor/output.hpp" #include "ngraph/descriptor/output.hpp"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment