Opensource Core ML delegate

coremltools 3.3 is added to third party dependencies.

PiperOrigin-RevId: 304316799
Change-Id: Ib14351512c2e57d7da70aa14991cd1b8354fcc03
This commit is contained in:
Taehee Jeong 2020-04-01 20:27:23 -07:00 committed by TensorFlower Gardener
parent 79e3d6dad5
commit df99d1db91
3 changed files with 26 additions and 0 deletions

View File

@ -32,6 +32,7 @@ tensorflow/third_party/clang_toolchain/download_clang.bzl
tensorflow/third_party/codegen.BUILD tensorflow/third_party/codegen.BUILD
tensorflow/third_party/com_google_absl.BUILD tensorflow/third_party/com_google_absl.BUILD
tensorflow/third_party/common.bzl tensorflow/third_party/common.bzl
tensorflow/third_party/coremltools.BUILD
tensorflow/third_party/cub.BUILD tensorflow/third_party/cub.BUILD
tensorflow/third_party/curl.BUILD tensorflow/third_party/curl.BUILD
tensorflow/third_party/cython.BUILD tensorflow/third_party/cython.BUILD

View File

@ -1045,6 +1045,16 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
"https://github.com/GrahamDumpleton/wrapt/archive/1.11.1.tar.gz", "https://github.com/GrahamDumpleton/wrapt/archive/1.11.1.tar.gz",
], ],
) )
tf_http_archive(
name = "coremltools",
sha256 = "0d594a714e8a5fd5bd740ad112ef59155c0482e25fdc8f8efa5758f90abdcf1e",
strip_prefix = "coremltools-3.3",
build_file = clean_dep("//third_party:coremltools.BUILD"),
urls = [
"http://mirror.tensorflow.org/github.com/apple/coremltools/archive/3.3.zip",
"https://github.com/apple/coremltools/archive/3.3.zip",
],
)
def tf_bind(): def tf_bind():
"""Bind targets for some external repositories""" """Bind targets for some external repositories"""

15
third_party/coremltools.BUILD vendored Normal file
View File

@ -0,0 +1,15 @@
load("@com_google_protobuf//:protobuf.bzl", "cc_proto_library")
package(default_visibility = ["//visibility:public"])
licenses(["notice"]) # BSD
exports_files(["LICENSE.txt"])
cc_proto_library(
name = "mlmodel_cc_proto",
srcs = glob(["mlmodel/format/*.proto"]),
include = "mlmodel/format",
default_runtime = "@com_google_protobuf//:protobuf_lite",
protoc = "@com_google_protobuf//:protoc",
)