Commit e17237de authored by pruthvi's avatar pruthvi

style fix

parent 3cd92bae
...@@ -17,9 +17,7 @@ ...@@ -17,9 +17,7 @@
#include "ngraph/runtime/cpu/cpu_mkl_allocator.hpp" #include "ngraph/runtime/cpu/cpu_mkl_allocator.hpp"
ngraph::runtime::cpu::CPUAllocator::CPUAllocator() ngraph::runtime::cpu::CPUAllocator::CPUAllocator()
: m_buffer(nullptr) : m_buffer(nullptr)
, m_byte_size(0) , m_byte_size(0)
{ {
} }
...@@ -51,23 +51,15 @@ namespace ngraph ...@@ -51,23 +51,15 @@ namespace ngraph
class ngraph::runtime::cpu::CPUAllocator class ngraph::runtime::cpu::CPUAllocator
{ {
public:
CPUAllocator(size_t size);
CPUAllocator();
~CPUAllocator();
public: private:
CPUAllocator(size_t size);
CPUAllocator();
~CPUAllocator();
private:
char* m_buffer; char* m_buffer;
size_t m_byte_size; size_t m_byte_size;
inline void* MallocHook(size_t size) inline void* MallocHook(size_t size) { m_buffer = static_cast<char*>(ngraph_malloc(size)); }
{ inline void FreeHook() { ngraph_free(m_buffer); }
m_buffer = static_cast<char*>(ngraph_malloc(size));
}
inline void FreeHook()
{
ngraph_free(m_buffer);
}
}; };
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