diff options
author | Nupur Garg <nupurgarg@google.com> | 2018-08-28 18:16:23 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-28 18:23:02 -0700 |
commit | 2e7352e57c541908cd700bb0fe53a04b456392c9 (patch) | |
tree | 2064e341d1a7f154d0cb1d42910359c7dd3e5a02 /tensorflow/workspace.bzl | |
parent | c4099e6ee8ba3846f2b7e70445806bc3055c5624 (diff) |
Add more model support to TocoConverter.
PiperOrigin-RevId: 210643904
Diffstat (limited to 'tensorflow/workspace.bzl')
-rwxr-xr-x | tensorflow/workspace.bzl | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 941b27cb59..de38f8c0c2 100755 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -767,6 +767,7 @@ def tf_workspace(path_prefix = "", tf_repo_name = ""): ], build_file = str(Label("//third_party:tflite_mobilenet.BUILD")), ) + tf_http_archive( name = "tflite_mobilenet_ssd_quant", sha256 = "a809cd290b4d6a2e8a9d5dad076e0bd695b8091974e0eed1052b480b2f21b6dc", @@ -778,6 +779,17 @@ def tf_workspace(path_prefix = "", tf_repo_name = ""): ) tf_http_archive( + name = "tflite_mobilenet_ssd_quant_protobuf", + sha256 = "09280972c5777f1aa775ef67cb4ac5d5ed21970acd8535aeca62450ef14f0d79", + urls = [ + "https://mirror.bazel.build/storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz", + "http://storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz", + ], + strip_prefix = "ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18", + build_file = str(Label("//third_party:tflite_mobilenet.BUILD")), + ) + + tf_http_archive( name = "tflite_conv_actions_frozen", sha256 = "d947b38cba389b5e2d0bfc3ea6cc49c784e187b41a071387b3742d1acac7691e", urls = [ |