Use .so or .dylib extension depending on the host os.

PiperOrigin-RevId: 258453048
Change-Id: Ie1f64eb1bfaa3e3f7cff53313a57edd75d9edd29

Author: ssjhv

workspace.bzl

@@ -21,15 +21,19 @@ def _tensorflow_pip_impl(ctx):
     fail("Failed to find include path. Did you remember to pip install " +
          "tensorflow?: %s" % include_path.stderr)
 
-  library_filename = "libtensorflow_framework.so.1"
+  if "linux" in ctx.os.name:
+    library_filename = "libtensorflow_framework.so.1"
+  elif "mac" in ctx.os.name:
+    library_filename = "libtensorflow_framework.dylib"
+
   ctx.symlink("/".join([library_path.stdout.strip(), library_filename]),
               library_filename)
   ctx.symlink(include_path.stdout.strip(), "include")
   ctx.file("BUILD", """
 cc_library(
     name = "libtensorflow_framework",
     srcs = ["{0}"],
-    hdrs = glob(["include/**/*.h"]),
+    hdrs = glob(["include/**"]),
     includes = ["include"],
     visibility = ["//visibility:public"],
 )