1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/framework/load_library.cc b/tensorflow/core/framework/load_library.cc
index f56e5fae1b..f825335300 100644
--- a/tensorflow/core/framework/load_library.cc
+++ b/tensorflow/core/framework/load_library.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include "tensorflow/core/framework/op_kernel.h"
 #include "tensorflow/core/lib/core/errors.h"
 #include "tensorflow/core/platform/env.h"
+#include "tensorflow/core/platform/mem.h"
 
 namespace tensorflow {
 
@@ -91,7 +92,7 @@ Status LoadLibrary(const char* library_filename, void** result,
   }
   string str;
   library.op_list.SerializeToString(&str);
-  char* str_buf = reinterpret_cast<char*>(malloc(str.length()));
+  char* str_buf = reinterpret_cast<char*>(port::Malloc(str.length()));
   memcpy(str_buf, str.data(), str.length());
   *buf = str_buf;
   *len = str.length();