WIP

src/ngraph/runtime/cpu/cpu_tensor_view.cpp

@@ -19,6 +19,7 @@
 #include "ngraph/shape.hpp"
 #include "ngraph/descriptor/layout/tensor_view_layout.hpp"
 #include "ngraph/descriptor/primary_tensor_view.hpp"
+#include "ngraph/runtime/cpu/cpu_layout_descriptor.hpp"
 #include "cpu_tensor_view.hpp"
 
 using namespace ngraph;
@@ -37,8 +38,12 @@ runtime::cpu::CPUTensorView::CPUTensorView(const ngraph::element::Type& element_
     , buffer(nullptr)
     , aligned_buffer(nullptr)
 {
-    // m_descriptor->set_tensor_view_layout(
-    //     std::make_shared<ngraph::descriptor::layout::DenseTensorViewLayout>(*m_descriptor));
+    // TODO(jmenon): A fallback layout should not be needed but is required
+    // because of how some unit test functionality is written (ex. 'backprop_derivative')
+    // This needs to be removed
+    m_descriptor->set_tensor_view_layout(
+        std::make_shared<runtime::cpu::LayoutDescriptor>(*m_descriptor,
+                                                         runtime::cpu::LayoutDescriptor::create_native_axis_order(shape.size())));
 
     buffer_size = shape_size(shape) * element_type.size();
     if (buffer_size)