Opensource Core ML delegate
coremltools 3.3 is added to third party dependencies. PiperOrigin-RevId: 304316799 Change-Id: Ib14351512c2e57d7da70aa14991cd1b8354fcc03
This commit is contained in:
parent
79e3d6dad5
commit
df99d1db91
|
@ -32,6 +32,7 @@ tensorflow/third_party/clang_toolchain/download_clang.bzl
|
|||
tensorflow/third_party/codegen.BUILD
|
||||
tensorflow/third_party/com_google_absl.BUILD
|
||||
tensorflow/third_party/common.bzl
|
||||
tensorflow/third_party/coremltools.BUILD
|
||||
tensorflow/third_party/cub.BUILD
|
||||
tensorflow/third_party/curl.BUILD
|
||||
tensorflow/third_party/cython.BUILD
|
||||
|
|
|
@ -1045,6 +1045,16 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
|
|||
"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():
|
||||
"""Bind targets for some external repositories"""
|
||||
|
|
|
@ -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",
|
||||
)
|
Loading…
Reference in New Issue