Commit 7217ea3c authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

WIP

parent 38abf879
...@@ -94,14 +94,14 @@ ...@@ -94,14 +94,14 @@
#include "ngraph/runtime/cpu/cpu_tensor_view.hpp" #include "ngraph/runtime/cpu/cpu_tensor_view.hpp"
#include "ngraph/runtime/cpu/ops/matmul_bias.hpp" #include "ngraph/runtime/cpu/ops/matmul_bias.hpp"
#include "ngraph/runtime/cpu/pass/cpu_layout.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 std;
using namespace ngraph; using namespace ngraph;
static const string s_output_dir = "cpu_codegen"; static const string s_output_dir = "cpu_codegen";
// Temporary Memory pool alignment // Temporary Memory Pool alignment
static const size_t MemoryPoolAlignment = 64; static const size_t MemoryPoolAlignment = 64;
class StaticInitializers class StaticInitializers
......
...@@ -34,7 +34,9 @@ namespace ngraph ...@@ -34,7 +34,9 @@ namespace ngraph
{ {
for (size_t i = 0; i < node->get_output_size(); ++i) 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; return false;
......
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