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

WIP

parent c405c3bc
......@@ -12,6 +12,7 @@
// See the License for the specific language governing permissions and
// ----------------------------------------------------------------------------
#include <cassert>
#include <cstdlib>
#include <fstream>
#include <memory>
......@@ -25,7 +26,6 @@
#include "ngraph/codegen/compiler.hpp"
#include "ngraph/codegen/execution_engine.hpp"
#include "ngraph/descriptor/input.hpp"
#include "ngraph/descriptor/layout/dense_tensor_view_layout.hpp"
#include "ngraph/descriptor/output.hpp"
#include "ngraph/descriptor/primary_tensor_view.hpp"
#include "ngraph/file_util.hpp"
......
......@@ -49,7 +49,7 @@ namespace ngraph
throw ngraph_error("Axis is out of bounds");
}
strides.emplace_back(shape[*it]);
strides.emplace_back(s);
s *= shape[*it];
}
std::reverse(strides.begin(), strides.end());
......
......@@ -34,6 +34,7 @@ namespace ngraph
public:
LayoutDescriptor(const ngraph::descriptor::TensorView& tv,
const AxisVector& tv_axis_order);
~LayoutDescriptor() { }
size_t get_size() override { return size; }
size_t get_offset() const { return offset; }
......
......@@ -13,6 +13,7 @@
// ----------------------------------------------------------------------------
#include <algorithm>
#include <memory>
#include "cpu_layout.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