WIP

src/ngraph/runtime/cpu/cpu_external_function.cpp

@@ -94,14 +94,14 @@
 #include "ngraph/runtime/cpu/cpu_tensor_view.hpp"
 #include "ngraph/runtime/cpu/ops/matmul_bias.hpp"
 #include "ngraph/runtime/cpu/pass/cpu_layout.hpp"
-#include "ngraph/runtime/cpu/pass/cpu_tensor_allocation.hpp"
+//#include "ngraph/runtime/cpu/pass/cpu_tensor_allocation.hpp"
 
 using namespace std;
 using namespace ngraph;
 
 static const string s_output_dir = "cpu_codegen";
 
-// Temporary Memory pool alignment
+// Temporary Memory Pool alignment
 static const size_t MemoryPoolAlignment = 64;
 
 class StaticInitializers

src/ngraph/runtime/cpu/pass/cpu_tensor_allocation.hpp

@@ -34,7 +34,9 @@ namespace ngraph
                     {
                         for (size_t i = 0; i < node->get_output_size(); ++i)
                         {
-                            auto tv = node->get_output_tensor_view(i);
+                            auto tensor_view = node->get_output_tensor_view(i);
+                            auto cpu_tensor_view = std::static_pointer_cast<ngraph::runtime::cpu::CPUTensorView>(tensor_view);
+
                         }
                     }
                     return false;