experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing MklCPUAllocator error introduced by commit #1baac7862739525351d25202800dc04e8ec3868b

Browse Source
This commit is contained in:
Clayne Robison 2018-02-13 20:07:42 -08:00
parent b2a0f1c45b
commit 95fae75e4d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/common_runtime/mkl_cpu_allocator.h
View File

@ -25,7 +25,7 @@ limitations under the License.
#include <cstdlib> #include <cstdlib>
#include <string> #include <string>
#include "tensorflow/core/common_runtime/bfc_allocator.h" #include "tensorflow/core/common_runtime/bfc_allocator.h"
#include "tensorflow/core/framework/allocator.h" #include "tensorflow/core/common_runtime/visitable_allocator.h"
#include "tensorflow/core/lib/strings/numbers.h" #include "tensorflow/core/lib/strings/numbers.h"
#include "tensorflow/core/lib/strings/str_util.h" #include "tensorflow/core/lib/strings/str_util.h"
#include "tensorflow/core/platform/mem.h" #include "tensorflow/core/platform/mem.h"
@ -161,7 +161,7 @@ class MklCPUAllocator : public VisitableAllocator {
/// The alignment that we need for the allocations /// The alignment that we need for the allocations
static const size_t kAlignment = 64; static const size_t kAlignment = 64;
Allocator* allocator_; // owned by this class VisitableAllocator* allocator_; // owned by this class
}; };
} // namespace tensorflow } // namespace tensorflow