Project import generated by Copybara.
GitOrigin-RevId: 1e13be30e2c6838d4a2ff768a39c414bc80534bb
This commit is contained in:
MediaPipe Team
committed by
Sebastian Schmidt
Sebastian Schmidt
parent
63e679d99c
commit
4dc4b19ddb
Vendored
+593
@@ -0,0 +1,593 @@
|
||||
"""
|
||||
External file definitions for MediaPipe.
|
||||
|
||||
This file is auto-generated.
|
||||
"""
|
||||
|
||||
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_file")
|
||||
|
||||
# buildifier: disable=unnamed-macro
|
||||
def external_files():
|
||||
"""External file definitions for MediaPipe."""
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_30k-clean_model",
|
||||
sha256 = "fefb02b667a6c5c2fe27602d28e5fb3428f66ab89c7d6f388e7c8d44a02d0336",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/30k-clean.model?generation=1661875643984613"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_albert_with_metadata_tflite",
|
||||
sha256 = "6012e264092d40a2e14f634579b95c6fa9938d7995de810e26fcec65cbcd6442",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/albert_with_metadata.tflite?generation=1661875651648830"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_bert_nl_classifier_tflite",
|
||||
sha256 = "1e5a550c09bff0a13e61858bcfac7654d7fcc6d42106b4f15e11117695069600",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/bert_nl_classifier.tflite?generation=1661875658827092"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_BUILD",
|
||||
sha256 = "d2b2a8346202691d7f831887c84e9642e974f64ed67851d9a58cf15c94b1f6b3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/BUILD?generation=1661875663693976"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_burger_jpg",
|
||||
sha256 = "97c15bbbf3cf3615063b1031c85d669de55839f59262bbe145d15ca75b36ecbf",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/burger.jpg?generation=1661875667922678"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_cat_jpg",
|
||||
sha256 = "2533197401eebe9410ea4d063f86c43fbd2666f3e8165a38aca155c0d09c21be",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/cat.jpg?generation=1661875672441459"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_cat_mask_jpg",
|
||||
sha256 = "bae065a685f2d32f1856151b5181671aa4d09925b55766935a30bbc8dafadcd0",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/cat_mask.jpg?generation=1661875677203533"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_cats_and_dogs_jpg",
|
||||
sha256 = "a2eaa7ad3a1aae4e623dd362a5f737e8a88d122597ecd1a02b3e1444db56df9c",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/cats_and_dogs.jpg?generation=1661875684064150"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_cats_and_dogs_no_resizing_jpg",
|
||||
sha256 = "9d55933ed66bcdc63cd6509ee2518d7eed75d12db609238387ee4cc50b173e58",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/cats_and_dogs_no_resizing.jpg?generation=1661875687251296"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_coco_efficientdet_lite0_v1_1_0_quant_2021_09_06_tflite",
|
||||
sha256 = "dee1b4af055a644804d5594442300ecc9e4f7080c25b7c044c98f527eeabb6cf",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/coco_efficientdet_lite0_v1_1.0_quant_2021_09_06.tflite?generation=1661875692679200"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_coco_ssd_mobilenet_v1_1_0_quant_2018_06_29_score_calibration_tflite",
|
||||
sha256 = "072b44c01f35ba4274adfab69bd8b0f21e7481168782279105426a25b6da5d4a",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29_score_calibration.tflite?generation=1661875697443279"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_coco_ssd_mobilenet_v1_1_0_quant_2018_06_29_tflite",
|
||||
sha256 = "61d598093ed03ed41aa47c3a39a28ac01e960d6a810a5419b9a5016a1e9c469b",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/coco_ssd_mobilenet_v1_1.0_quant_2018_06_29.tflite?generation=1661875702588267"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_corrupted_mobilenet_v1_0_25_224_1_default_1_tflite",
|
||||
sha256 = "f0cbeb8061f4c693e20de779ce255af923508492e8a24f6db320845a52facb51",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/corrupted_mobilenet_v1_0.25_224_1_default_1.tflite?generation=1661875706780536"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_deeplabv3_tflite",
|
||||
sha256 = "9711334db2b01d5894feb8ed0f5cb3e97d125b8d229f8d8692f625801818f5ef",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/deeplabv3.tflite?generation=1661875711618421"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_empty_vocab_for_regex_tokenizer_txt",
|
||||
sha256 = "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/empty_vocab_for_regex_tokenizer.txt?generation=1661875714907539"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_expected_left_down_hand_landmarks_prototxt",
|
||||
sha256 = "ae9cb01035f18b0023fc12256c048666da76b41b327cec09c2d2820054b1295f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/expected_left_down_hand_landmarks.prototxt?generation=1661875720230540"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_expected_left_up_hand_landmarks_prototxt",
|
||||
sha256 = "1353ba617c4f048083618587cd23a8a22115f634521c153d4e1bd1ebd4f49dd7",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/expected_left_up_hand_landmarks.prototxt?generation=1661875726008879"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_expected_right_down_hand_landmarks_prototxt",
|
||||
sha256 = "f281b745175aaa7f458def6cf4c89521fb56302dd61a05642b3b4a4f237ffaa3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/expected_right_down_hand_landmarks.prototxt?generation=1661875730821226"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_expected_right_up_hand_landmarks_prototxt",
|
||||
sha256 = "174cf5f7c3ab547f0affb666ee7be933b0758c60fbfe7b7e93795c5082555592",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/expected_right_up_hand_landmarks.prototxt?generation=1661875733440313"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_external_file_txt",
|
||||
sha256 = "ae0666f161fed1a5dde998bbd0e140550d2da0db27db1d0e31e370f2bd366a57",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/external_file.txt?generation=1661875736240688"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_face_detection_full_range_sparse_tflite",
|
||||
sha256 = "671dd2f9ed11a78436fc21cc42357a803dfc6f73e9fb86541be942d5716c2dce",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/face_detection_full_range_sparse.tflite?generation=1661875739104017"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_face_detection_full_range_tflite",
|
||||
sha256 = "99bf9494d84f50acc6617d89873f71bf6635a841ea699c17cb3377f9507cfec3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/face_detection_full_range.tflite?generation=1661875742733283"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_face_detection_short_range_tflite",
|
||||
sha256 = "3bc182eb9f33925d9e58b5c8d59308a760f4adea8f282370e428c51212c26633",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/face_detection_short_range.tflite?generation=1661875748538815"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_face_landmark_tflite",
|
||||
sha256 = "c603fa6149219a3e9487dc9abd7a0c24474c77263273d24868378cdf40aa26d1",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/face_landmark.tflite?generation=1662063817995673"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_face_landmark_with_attention_tflite",
|
||||
sha256 = "883b7411747bac657c30c462d305d312e9dec6adbf8b85e2f5d8d722fca9455d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/face_landmark_with_attention.tflite?generation=1661875751615925"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_hair_segmentation_tflite",
|
||||
sha256 = "d2c940c4fd80edeaf38f5d7387d1b4235ee320ed120080df67c663e749e77633",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/hair_segmentation.tflite?generation=1661875756623461"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_hand_landmark_full_tflite",
|
||||
sha256 = "11c272b891e1a99ab034208e23937a8008388cf11ed2a9d776ed3d01d0ba00e3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/hand_landmark_full.tflite?generation=1661875760968579"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_hand_landmark_lite_tflite",
|
||||
sha256 = "048edd3645c9bf7397d19a9f6e3a42957d6e414c9bea6598030a2e9b624156e6",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/hand_landmark_lite.tflite?generation=1661875766398729"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_hand_recrop_tflite",
|
||||
sha256 = "67d996ce96f9d36fe17d2693022c6da93168026ab2f028f9e2365398d8ac7d5d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/hand_recrop.tflite?generation=1661875770633070"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_iris_and_gaze_tflite",
|
||||
sha256 = "b6dcb860a92a3c7264a8e50786f46cecb529672cdafc17d39c78931257da661d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/iris_and_gaze.tflite?generation=1661875774291949"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_iris_landmark_tflite",
|
||||
sha256 = "d1744d2a09c25f501d39eba4faff47e53ecca8852c5ce19bce8eeac39357521f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/iris_landmark.tflite?generation=1662065468653224"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_knift_float_1k_tflite",
|
||||
sha256 = "5dbfa98c7a3caae97840576a278a1d1fe37c86bad4007d1acdffec094242837c",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/knift_float_1k.tflite?generation=1661875777483362"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_knift_float_400_tflite",
|
||||
sha256 = "3ee576050f3d5d45ea19a19dbd67267cb345b0348efde00952eddb8b7aabe2e5",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/knift_float_400.tflite?generation=1661875782481333"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_knift_float_tflite",
|
||||
sha256 = "40567854c2c1022c98cd2c55a7eef1c60999580ce67db118c1274000d0e22ace",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/knift_float.tflite?generation=1661875785348544"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_knift_index_pb",
|
||||
sha256 = "2c2b57a846e0adbf1e3f25bd20c7878ac9399460a1ad5d8147e3231ace8eb3dc",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/knift_index.pb?generation=1661875789855286"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_knift_labelmap_txt",
|
||||
sha256 = "40f9f5bd76a8574478299af93fcab96f5cdc71273f4e20c5899c248a33970cff",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/knift_labelmap.txt?generation=1661875792821628"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_left_hands_jpg",
|
||||
sha256 = "4b5134daa4cb60465535239535f9f74c2842aba3aa5fd30bf04ef5678f93d87f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/left_hands.jpg?generation=1661875796949017"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilebert_vocab_txt",
|
||||
sha256 = "07eced375cec144d27c900241f3e339478dec958f92fddbc551f295c992038a3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilebert_vocab.txt?generation=1661875800701493"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilebert_with_metadata_tflite",
|
||||
sha256 = "5984e86eb5d4cb95f004ff78e6f44d5f59b17120575c6313955d95afbb843ca3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilebert_with_metadata.tflite?generation=1661875806733025"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobile_ica_8bit-with-metadata_tflite",
|
||||
sha256 = "4afa3970d3efd6726d147d505e28c7ff1e4fe1c24be7bcda6b5429eb099777a5",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobile_ica_8bit-with-metadata.tflite?generation=1661875810860490"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobile_ica_8bit-without-model-metadata_tflite",
|
||||
sha256 = "407d7b11da4b9e3f56f0cff7075e86a3d70813c74a15cf11975176912c65cbde",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobile_ica_8bit-without-model-metadata.tflite?generation=1661875814428283"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobile_ica_8bit-with-unsupported-metadata-version_tflite",
|
||||
sha256 = "5ea0341c481367df51741d7aa2fab4e3ba59f67ab366b18f6dcd50cb859ed548",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobile_ica_8bit-with-unsupported-metadata-version.tflite?generation=1661875819091013"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v1_0_25_192_quantized_1_default_1_tflite",
|
||||
sha256 = "f80999b6324c6f101300c3ee38fbe7e11e74a743b5e0be7350602087fe7430a3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v1_0.25_192_quantized_1_default_1.tflite?generation=1661875821863721"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v1_0_25_224_1_default_1_tflite",
|
||||
sha256 = "446ec673881cd46371a8726075b714194ada39d144762260cb76d15318597df7",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v1_0.25_224_1_default_1.tflite?generation=1661875824782010"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v1_0_25_224_1_metadata_1_tflite",
|
||||
sha256 = "348cc1221740b9fe1f609c964eff5bf09650bda76341c30aa27800b3da6171f4",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v1_0.25_224_1_metadata_1.tflite?generation=1661875828385370"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v1_0_25_224_quant_tflite",
|
||||
sha256 = "e480eb15572f86d3d5f1be6e83e35b3c7d509ab2bcec353707d1f614e14edca2",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v1_0.25_224_quant.tflite?generation=1661875831485992"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v1_0_25_224_quant_without_subgraph_metadata_tflite",
|
||||
sha256 = "78f8b9bb5c873d3ad53ffc03b27651213016e45b6a2df42010c93191293bf694",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v1_0.25_224_quant_without_subgraph_metadata.tflite?generation=1661875836078124"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobilenet_v2_1_0_224_tflite",
|
||||
sha256 = "ff5cb7f9e62c92ebdad971f8a98aa6b3106d82a64587a7787c6a385c9e791339",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobilenet_v2_1.0_224.tflite?generation=1661875840611150"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobile_object_classifier_v0_2_3-metadata-no-name_tflite",
|
||||
sha256 = "27fdb2dce68b8bd9a0f16583eefc4df13605808c1417cec268d1e838920c1a81",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobile_object_classifier_v0_2_3-metadata-no-name.tflite?generation=1661875843557142"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mobile_object_labeler_v1_tflite",
|
||||
sha256 = "9400671e04685f5277edd3052a311cc51533de9da94255c52ebde1e18484c77c",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mobile_object_labeler_v1.tflite?generation=1661875846924538"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_model_without_metadata_tflite",
|
||||
sha256 = "05c5aea7ae00aeed0053a85f2b2e896b4ea272c5219052d32c06b655fbf5cc9b",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/model_without_metadata.tflite?generation=1661875850966737"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_mozart_square_jpg",
|
||||
sha256 = "4feb4dadc5d6f853ade57b8c9d4c9a1f5ececd6469616c8e505f9a14823392b6",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/mozart_square.jpg?generation=1661875853838871"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_camera_tflite",
|
||||
sha256 = "f66e92e81ed3f4698f74d565a7668e016e2288ea92fb42938e33b778bd1e110d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_camera.tflite?generation=1661875857210211"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_chair_1stage_tflite",
|
||||
sha256 = "694af9bdcea270f2bad488beb4e5ef89aad819489d5d9aa4a774d2fad2a91ae9",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_chair_1stage.tflite?generation=1661875860251330"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_chair_tflite",
|
||||
sha256 = "190e4ea49ba891ed242ddc73703e03d70164c27f3da07492d7010379e24f2a6b",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_chair.tflite?generation=1661875863685724"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_cup_tflite",
|
||||
sha256 = "c4f4ea8def16bd191d11279f754e6f3f2a9d94839a956b975e5697e943157ac7",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_cup.tflite?generation=1661875867924057"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_sneakers_1stage_tflite",
|
||||
sha256 = "ef052353e882d93429ee90a8e8e5e781f04acdf44c0cef4d961d8cbfa89aad8c",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_sneakers_1stage.tflite?generation=1661875871321513"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_3d_sneakers_tflite",
|
||||
sha256 = "4eb1633d646a43ae979ba497487e95dbf89f97406ed02200ae39ae46b0a0543d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_3d_sneakers.tflite?generation=1661875875616135"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_ssd_mobilenetv2_oidv4_fp16_tflite",
|
||||
sha256 = "d0a5255bf8c4f5a0bc4240741a76c41d5e939f7655078f945f50ab53a9375da6",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_ssd_mobilenetv2_oidv4_fp16.tflite?generation=1661875879063676"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_palm_detection_full_tflite",
|
||||
sha256 = "2f25e740121983f68ffc05f99991d524dc0ea812134f6316a26125816941ee85",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/palm_detection_full.tflite?generation=1661875883244842"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_palm_detection_lite_tflite",
|
||||
sha256 = "e9a4aaddf90dda56a87235303cf00e4c2d3fb28725f68fd88772997dac905c18",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/palm_detection_lite.tflite?generation=1661875885885770"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_pose_detection_tflite",
|
||||
sha256 = "a63c614bef30d35947f13be361820b1e4e3bec9cfeebf4d11216a18373108e85",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/pose_detection.tflite?generation=1661875889147923"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_pose_landmark_full_tflite",
|
||||
sha256 = "e9a5c5cb17f736fafd4c2ec1da3b3d331d6edbe8a0d32395855aeb2cdfd64b9f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/pose_landmark_full.tflite?generation=1661875894245786"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_pose_landmark_heavy_tflite",
|
||||
sha256 = "59e42d71bcd44cbdbabc419f0ff76686595fd265419566bd4009ef703ea8e1fe",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/pose_landmark_heavy.tflite?generation=1661875897944151"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_pose_landmark_lite_tflite",
|
||||
sha256 = "f17bfbecadb61c3be1baa8b8d851cc6619c870a87167b32848ad20db306b9d61",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/pose_landmark_lite.tflite?generation=1661875901231143"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_README_md",
|
||||
sha256 = "a96d08c9c70cd9717207ed72c926e02e5eada751f00bdc5d3a7e82e3492b72cb",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/README.md?generation=1661875904887163"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_right_hands_jpg",
|
||||
sha256 = "240c082e80128ff1ca8a83ce645e2ba4d8bc30f0967b7991cf5fa375bab489e1",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/right_hands.jpg?generation=1661875908672404"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_segmentation_golden_rotation0_png",
|
||||
sha256 = "9ee993919b753118928ba2d14f7c5c83a6cfc23355e6943dac4ad81eedd73069",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/segmentation_golden_rotation0.png?generation=1661875911319083"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_segmentation_input_rotation0_jpg",
|
||||
sha256 = "5bf58d8af1f1c33224f3f3bc0ce451c8daf0739cc15a86d59d8c3bf2879afb97",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/segmentation_input_rotation0.jpg?generation=1661875914048401"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segm_128_128_3_expected_mask_jpg",
|
||||
sha256 = "a295f3ab394a5e0caff2db5041337da58341ec331f1413ef91f56e0d650b4a1e",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segm_128_128_3_expected_mask.jpg?generation=1661875916766416"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segm_128_128_3_tflite",
|
||||
sha256 = "bb154f248543c0738e32f1c74375245651351a84746dc21f10bdfaabd8fae4ca",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segm_128_128_3.tflite?generation=1661875919964123"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segm_144_256_3_expected_mask_jpg",
|
||||
sha256 = "cfc699db9670585c04414d0d1a07b289a027ba99d6903d2219f897d34e2c9952",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segm_144_256_3_expected_mask.jpg?generation=1661875922646736"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segm_144_256_3_tflite",
|
||||
sha256 = "5c770b8834ad50586599eae7710921be09d356898413fc0bf37a9458da0610eb",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segm_144_256_3.tflite?generation=1661875925519713"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segmentation_landscape_tflite",
|
||||
sha256 = "4aafe6223bb8dac6fac8ca8ed56852870a33051ef3f6238822d282a109962894",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segmentation_landscape.tflite?generation=1661875928328455"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_selfie_segmentation_tflite",
|
||||
sha256 = "8d13b7fae74af625c641226813616a2117bd6bca19eb3b75574621fc08557f27",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/selfie_segmentation.tflite?generation=1661875931201364"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_speech_16000_hz_mono_wav",
|
||||
sha256 = "71caf50b8757d6ab9cad5eae4d36669d3c20c225a51660afd7fe0dc44cdb74f6",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/speech_16000_hz_mono.wav?generation=1661875934539524"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_speech_48000_hz_mono_wav",
|
||||
sha256 = "04d4590b61d0519170d7aa0686ab2ff5da2b8487d192e40413dd36d9c1a24304",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/speech_48000_hz_mono.wav?generation=1661875938066405"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_ssdlite_object_detection_labelmap_txt",
|
||||
sha256 = "c7e79c855f73cbba9f33d649d60e1676eb0a974021a41696d1ac0d4b7f7e0211",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/ssdlite_object_detection_labelmap.txt?generation=1661875940778557"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_ssdlite_object_detection_tflite",
|
||||
sha256 = "8e10a2e2f5db85d8f90628f00752a89ff241c5b2ca82f3b92fc496c7bda122ef",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/ssdlite_object_detection.tflite?generation=1661875944118759"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_ssd_mobilenet_v1_tflite",
|
||||
sha256 = "cbdecd08b44c5dea3821f77c5468e2936ecfbf43cde0795a2729fdb43401e58b",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/ssd_mobilenet_v1.tflite?generation=1661875947436302"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_test_model_add_op_tflite",
|
||||
sha256 = "298300ca8a9193b80ada1dca39d36f20bffeebde09e85385049b3bfe7be2272f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/test_model_add_op.tflite?generation=1661875950076192"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_test_model_nl_classifier_with_regex_tokenizer_tflite",
|
||||
sha256 = "cb12618d084b813cb7b90ceb39c9fe4b18dae4de9880b912cdcd4b577cd65b4f",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/test_model_nl_classifier_with_regex_tokenizer.tflite?generation=1661875953222362"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_test_model_with_custom_op_tflite",
|
||||
sha256 = "bafff7c8508ac24846e089ab70dcf48943a483a3e20290ff60e7740d073d7653",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/test_model_with_custom_op.tflite?generation=1661875957061036"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_test_model_without_custom_op_tflite",
|
||||
sha256 = "e17f0a1a22bc9242d9f825fe1edce07d2f90eb2a57e8b29a996244f194ee08a0",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/test_model_without_custom_op.tflite?generation=1661875959757731"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_two_heads_16000_hz_mono_wav",
|
||||
sha256 = "a291a9c22c39bba30138a26915e154a96286ba6ca3b413053123c504a58cce3b",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/two_heads_16000_hz_mono.wav?generation=1661875962421337"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_two_heads_44100_hz_mono_wav",
|
||||
sha256 = "1bf525ad7b7bac2da65addb5593b49adaba52ec3a9ed891f70afe0b392db02cd",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/two_heads_44100_hz_mono.wav?generation=1661875965316595"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_two_heads_tflite",
|
||||
sha256 = "bfa6ee4ccaf9180b69b39fa579b26b74bbf7758ae398e1d2265a58d323ca3d84",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/two_heads.tflite?generation=1661875968723352"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_vocab_for_regex_tokenizer_txt",
|
||||
sha256 = "b1134b10927a53ce4224bbc30ccf075c9969c94ebf40c368966d1dcf445ca923",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/vocab_for_regex_tokenizer.txt?generation=1661875971574893"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_vocab_txt",
|
||||
sha256 = "a125f531f48943ac4c3f117112150b91825aed560d890718dd96dc764a2bc141",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/vocab.txt?generation=1661875974626008"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_vocab_with_index_txt",
|
||||
sha256 = "664d78a2835bba781c23f9b556886bfcd8eef3d2a7414cf31d5c6963d9669379",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/vocab_with_index.txt?generation=1661875977280658"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_yamnet_audio_classifier_with_metadata_tflite",
|
||||
sha256 = "10c95ea3eb9a7bb4cb8bddf6feb023250381008177ac162ce169694d05c317de",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/yamnet_audio_classifier_with_metadata.tflite?generation=1661875980774466"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_model_ckpt_data-00000-of-00001",
|
||||
sha256 = "ad2f733f271dd5000a8c7f926bfea1083e6408b34d4f3b60679e5a6f96251c97",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/model.ckpt.data-00000-of-00001?generation=1661875984176294"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_model_ckpt_index",
|
||||
sha256 = "283816fcab228e6246d1c03b596f50dd40e4fe3e04c52a522a5b9d6f2cc43273",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/model.ckpt.index?generation=1661875987100245"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_model_ckpt_meta",
|
||||
sha256 = "9d80696ab76a492a23f6ce1d0d33b2d13c26e118b86d3ef61b691ad67d0f1f5a",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/model.ckpt.meta?generation=1661875990332395"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_pipeline_config",
|
||||
sha256 = "995aff0b28af5f66eb98d0734494395710ae84c843aee207755e7bc5025c9abb",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/pipeline.config?generation=1661875993079273"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_README_md",
|
||||
sha256 = "fe163cf12fbd017738a2fd360c03d223e964ba6404ac75c635f5918784e9c34d",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/README.md?generation=1661875995856372"],
|
||||
)
|
||||
|
||||
http_file(
|
||||
name = "com_google_mediapipe_object_detection_saved_model_saved_model_pb",
|
||||
sha256 = "f29606cf218397d5580c496e50fd28cddf66e2f59b819ab9c761b72270a5adf3",
|
||||
urls = ["https://storage.googleapis.com/mediapipe-assets/object_detection_saved_model/saved_model.pb?generation=1661875999264354"],
|
||||
)
|
||||
Vendored
+1
@@ -0,0 +1 @@
|
||||
# This empty BUILD file is required to make Bazel treat this directory as a package.
|
||||
Vendored
+191
@@ -0,0 +1,191 @@
|
||||
load("@build_bazel_rules_android//android:rules.bzl", "android_library")
|
||||
load(":build_defs.bzl", "flatbuffer_py_strip_prefix_srcs")
|
||||
load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_library")
|
||||
|
||||
package(default_visibility = ["//visibility:public"])
|
||||
|
||||
licenses(["notice"]) # Apache 2.0
|
||||
|
||||
exports_files(["LICENSE.txt"])
|
||||
|
||||
licenses(["notice"])
|
||||
|
||||
config_setting(
|
||||
name = "platform_freebsd",
|
||||
values = {"cpu": "freebsd"},
|
||||
)
|
||||
|
||||
config_setting(
|
||||
name = "platform_openbsd",
|
||||
values = {"cpu": "openbsd"},
|
||||
)
|
||||
|
||||
config_setting(
|
||||
name = "windows",
|
||||
values = {"cpu": "x64_windows"},
|
||||
)
|
||||
|
||||
# Public flatc library to compile flatbuffer files at runtime.
|
||||
cc_library(
|
||||
name = "flatbuffers",
|
||||
hdrs = ["//:public_headers"],
|
||||
linkstatic = 1,
|
||||
strip_include_prefix = "/include",
|
||||
visibility = ["//visibility:public"],
|
||||
deps = ["//src:flatbuffers"],
|
||||
)
|
||||
|
||||
# Public C++ headers for the Flatbuffers library.
|
||||
filegroup(
|
||||
name = "public_headers",
|
||||
srcs = [
|
||||
"include/flatbuffers/allocator.h",
|
||||
"include/flatbuffers/array.h",
|
||||
"include/flatbuffers/base.h",
|
||||
"include/flatbuffers/bfbs_generator.h",
|
||||
"include/flatbuffers/buffer.h",
|
||||
"include/flatbuffers/buffer_ref.h",
|
||||
"include/flatbuffers/code_generators.h",
|
||||
"include/flatbuffers/default_allocator.h",
|
||||
"include/flatbuffers/detached_buffer.h",
|
||||
"include/flatbuffers/flatbuffer_builder.h",
|
||||
"include/flatbuffers/flatbuffers.h",
|
||||
"include/flatbuffers/flexbuffers.h",
|
||||
"include/flatbuffers/hash.h",
|
||||
"include/flatbuffers/idl.h",
|
||||
"include/flatbuffers/minireflect.h",
|
||||
"include/flatbuffers/reflection.h",
|
||||
"include/flatbuffers/reflection_generated.h",
|
||||
"include/flatbuffers/registry.h",
|
||||
"include/flatbuffers/stl_emulation.h",
|
||||
"include/flatbuffers/string.h",
|
||||
"include/flatbuffers/struct.h",
|
||||
"include/flatbuffers/table.h",
|
||||
"include/flatbuffers/util.h",
|
||||
"include/flatbuffers/vector.h",
|
||||
"include/flatbuffers/vector_downward.h",
|
||||
"include/flatbuffers/verifier.h",
|
||||
],
|
||||
visibility = ["//:__subpackages__"],
|
||||
)
|
||||
|
||||
# Public flatc compiler library.
|
||||
cc_library(
|
||||
name = "flatc_library",
|
||||
linkstatic = 1,
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
"@flatbuffers//src:flatc_library",
|
||||
],
|
||||
)
|
||||
|
||||
# Public flatc compiler.
|
||||
cc_binary(
|
||||
name = "flatc",
|
||||
linkopts = select({
|
||||
":platform_freebsd": [
|
||||
"-lm",
|
||||
],
|
||||
":windows": [],
|
||||
"//conditions:default": [
|
||||
"-lm",
|
||||
"-ldl",
|
||||
],
|
||||
}),
|
||||
visibility = ["//visibility:public"],
|
||||
deps = [
|
||||
"@flatbuffers//src:flatc",
|
||||
],
|
||||
)
|
||||
|
||||
filegroup(
|
||||
name = "flatc_headers",
|
||||
srcs = [
|
||||
"include/flatbuffers/flatc.h",
|
||||
],
|
||||
visibility = ["//:__subpackages__"],
|
||||
)
|
||||
|
||||
# Library used by flatbuffer_cc_library rules.
|
||||
cc_library(
|
||||
name = "runtime_cc",
|
||||
hdrs = [
|
||||
"include/flatbuffers/allocator.h",
|
||||
"include/flatbuffers/array.h",
|
||||
"include/flatbuffers/base.h",
|
||||
"include/flatbuffers/buffer.h",
|
||||
"include/flatbuffers/buffer_ref.h",
|
||||
"include/flatbuffers/default_allocator.h",
|
||||
"include/flatbuffers/detached_buffer.h",
|
||||
"include/flatbuffers/flatbuffer_builder.h",
|
||||
"include/flatbuffers/flatbuffers.h",
|
||||
"include/flatbuffers/flexbuffers.h",
|
||||
"include/flatbuffers/stl_emulation.h",
|
||||
"include/flatbuffers/string.h",
|
||||
"include/flatbuffers/struct.h",
|
||||
"include/flatbuffers/table.h",
|
||||
"include/flatbuffers/util.h",
|
||||
"include/flatbuffers/vector.h",
|
||||
"include/flatbuffers/vector_downward.h",
|
||||
"include/flatbuffers/verifier.h",
|
||||
],
|
||||
linkstatic = 1,
|
||||
strip_include_prefix = "/include",
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
||||
flatbuffer_py_strip_prefix_srcs(
|
||||
name = "flatbuffer_py_strip_prefix",
|
||||
srcs = [
|
||||
"python/flatbuffers/__init__.py",
|
||||
"python/flatbuffers/_version.py",
|
||||
"python/flatbuffers/builder.py",
|
||||
"python/flatbuffers/compat.py",
|
||||
"python/flatbuffers/encode.py",
|
||||
"python/flatbuffers/flexbuffers.py",
|
||||
"python/flatbuffers/number_types.py",
|
||||
"python/flatbuffers/packer.py",
|
||||
"python/flatbuffers/table.py",
|
||||
"python/flatbuffers/util.py",
|
||||
],
|
||||
strip_prefix = "python/flatbuffers/",
|
||||
)
|
||||
|
||||
filegroup(
|
||||
name = "runtime_py_srcs",
|
||||
srcs = [
|
||||
"__init__.py",
|
||||
"_version.py",
|
||||
"builder.py",
|
||||
"compat.py",
|
||||
"encode.py",
|
||||
"flexbuffers.py",
|
||||
"number_types.py",
|
||||
"packer.py",
|
||||
"table.py",
|
||||
"util.py",
|
||||
],
|
||||
)
|
||||
|
||||
py_library(
|
||||
name = "runtime_py",
|
||||
srcs = [":runtime_py_srcs"],
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
||||
filegroup(
|
||||
name = "runtime_java_srcs",
|
||||
srcs = glob(["java/com/google/flatbuffers/**/*.java"]),
|
||||
)
|
||||
|
||||
java_library(
|
||||
name = "runtime_java",
|
||||
srcs = [":runtime_java_srcs"],
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
|
||||
android_library(
|
||||
name = "runtime_android",
|
||||
srcs = [":runtime_java_srcs"],
|
||||
visibility = ["//visibility:public"],
|
||||
)
|
||||
+644
@@ -0,0 +1,644 @@
|
||||
"""BUILD rules for generating flatbuffer files."""
|
||||
|
||||
load("@build_bazel_rules_android//android:rules.bzl", "android_library")
|
||||
|
||||
flatc_path = "@flatbuffers//:flatc"
|
||||
zip_files = "@org_tensorflow//tensorflow_lite_support/tools:zip_files"
|
||||
|
||||
DEFAULT_INCLUDE_PATHS = [
|
||||
"./",
|
||||
"$(GENDIR)",
|
||||
"$(BINDIR)",
|
||||
]
|
||||
|
||||
DEFAULT_FLATC_ARGS = [
|
||||
"--no-union-value-namespacing",
|
||||
"--gen-object-api",
|
||||
]
|
||||
|
||||
def flatbuffer_library_public(
|
||||
name,
|
||||
srcs,
|
||||
outs,
|
||||
language_flag,
|
||||
out_prefix = "",
|
||||
includes = [],
|
||||
include_paths = [],
|
||||
compatible_with = [],
|
||||
flatc_args = DEFAULT_FLATC_ARGS,
|
||||
reflection_name = "",
|
||||
reflection_visibility = None, # buildifier: disable=unused-variable
|
||||
output_to_bindir = False):
|
||||
"""Generates code files for reading/writing the given flatbuffers in the requested language using the public compiler.
|
||||
|
||||
Outs:
|
||||
filegroup(name): all generated source files.
|
||||
Fileset([reflection_name]): (Optional) all generated reflection binaries.
|
||||
|
||||
Args:
|
||||
name: Rule name.
|
||||
srcs: Source .fbs files. Sent in order to the compiler.
|
||||
outs: Output files from flatc.
|
||||
language_flag: Target language flag. One of [-c, -j, -js].
|
||||
out_prefix: Prepend this path to the front of all generated files except on
|
||||
single source targets. Usually is a directory name.
|
||||
includes: Optional, list of filegroups of schemas that the srcs depend on.
|
||||
include_paths: Optional, list of paths the includes files can be found in.
|
||||
compatible_with: Optional, passed to genrule for environments this rule
|
||||
can be built for.
|
||||
flatc_args: Optional, list of additional arguments to pass to flatc.
|
||||
reflection_name: Optional, if set this will generate the flatbuffer
|
||||
reflection binaries for the schemas.
|
||||
reflection_visibility: The visibility of the generated reflection Fileset.
|
||||
output_to_bindir: Passed to genrule for output to bin directory.
|
||||
"""
|
||||
include_paths_cmd = ["-I %s" % (s) for s in include_paths]
|
||||
|
||||
# '$(@D)' when given a single source target will give the appropriate
|
||||
# directory. Appending 'out_prefix' is only necessary when given a build
|
||||
# target with multiple sources.
|
||||
output_directory = (
|
||||
("-o $(@D)/%s" % (out_prefix)) if len(srcs) > 1 else ("-o $(@D)")
|
||||
)
|
||||
genrule_cmd = " ".join([
|
||||
"for f in $(SRCS); do",
|
||||
"$(location %s)" % (flatc_path),
|
||||
" ".join(flatc_args),
|
||||
" ".join(include_paths_cmd),
|
||||
language_flag,
|
||||
output_directory,
|
||||
"$$f;",
|
||||
"done",
|
||||
])
|
||||
native.genrule(
|
||||
name = name,
|
||||
srcs = srcs,
|
||||
outs = outs,
|
||||
output_to_bindir = output_to_bindir,
|
||||
compatible_with = compatible_with,
|
||||
tools = includes + [flatc_path],
|
||||
cmd = genrule_cmd,
|
||||
message = "Generating flatbuffer files for %s:" % (name),
|
||||
)
|
||||
if reflection_name:
|
||||
reflection_genrule_cmd = " ".join([
|
||||
"for f in $(SRCS); do",
|
||||
"$(location %s)" % (flatc_path),
|
||||
"-b --schema",
|
||||
" ".join(flatc_args),
|
||||
" ".join(include_paths_cmd),
|
||||
language_flag,
|
||||
output_directory,
|
||||
"$$f;",
|
||||
"done",
|
||||
])
|
||||
reflection_outs = [
|
||||
(out_prefix + "%s.bfbs") % (s.replace(".fbs", "").split("/")[-1])
|
||||
for s in srcs
|
||||
]
|
||||
native.genrule(
|
||||
name = "%s_srcs" % reflection_name,
|
||||
srcs = srcs,
|
||||
outs = reflection_outs,
|
||||
output_to_bindir = output_to_bindir,
|
||||
compatible_with = compatible_with,
|
||||
tools = includes + [flatc_path],
|
||||
cmd = reflection_genrule_cmd,
|
||||
message = "Generating flatbuffer reflection binary for %s:" % (name),
|
||||
)
|
||||
# TODO: Make bazel rules proper and supported by flatbuffer
|
||||
# Have to comment this since FilesetEntry is not supported in bazel
|
||||
# starlark.
|
||||
# native.Fileset(
|
||||
# name = reflection_name,
|
||||
# out = "%s_out" % reflection_name,
|
||||
# entries = [
|
||||
# native.FilesetEntry(files = reflection_outs),
|
||||
# ],
|
||||
# visibility = reflection_visibility,
|
||||
# compatible_with = compatible_with,
|
||||
# )
|
||||
|
||||
def flatbuffer_cc_library(
|
||||
name,
|
||||
srcs,
|
||||
srcs_filegroup_name = "",
|
||||
out_prefix = "",
|
||||
includes = [],
|
||||
include_paths = [],
|
||||
compatible_with = [],
|
||||
flatc_args = DEFAULT_FLATC_ARGS,
|
||||
visibility = None,
|
||||
srcs_filegroup_visibility = None,
|
||||
gen_reflections = False):
|
||||
'''A cc_library with the generated reader/writers for the given flatbuffer definitions.
|
||||
|
||||
Outs:
|
||||
filegroup([name]_srcs): all generated .h files.
|
||||
filegroup(srcs_filegroup_name if specified, or [name]_includes if not):
|
||||
Other flatbuffer_cc_library's can pass this in for their `includes`
|
||||
parameter, if they depend on the schemas in this library.
|
||||
Fileset([name]_reflection): (Optional) all generated reflection binaries.
|
||||
cc_library([name]): library with sources and flatbuffers deps.
|
||||
|
||||
Remarks:
|
||||
** Because the genrule used to call flatc does not have any trivial way of
|
||||
computing the output list of files transitively generated by includes and
|
||||
--gen-includes (the default) being defined for flatc, the --gen-includes
|
||||
flag will not work as expected. The way around this is to add a dependency
|
||||
to the flatbuffer_cc_library defined alongside the flatc included Fileset.
|
||||
For example you might define:
|
||||
|
||||
flatbuffer_cc_library(
|
||||
name = "my_fbs",
|
||||
srcs = [ "schemas/foo.fbs" ],
|
||||
includes = [ "//third_party/bazz:bazz_fbs_includes" ],
|
||||
)
|
||||
|
||||
In which foo.fbs includes a few files from the Fileset defined at
|
||||
//third_party/bazz:bazz_fbs_includes. When compiling the library that
|
||||
includes foo_generated.h, and therefore has my_fbs as a dependency, it
|
||||
will fail to find any of the bazz *_generated.h files unless you also
|
||||
add bazz's flatbuffer_cc_library to your own dependency list, e.g.:
|
||||
|
||||
cc_library(
|
||||
name = "my_lib",
|
||||
deps = [
|
||||
":my_fbs",
|
||||
"//third_party/bazz:bazz_fbs"
|
||||
],
|
||||
)
|
||||
|
||||
Happy dependent Flatbuffering!
|
||||
|
||||
Args:
|
||||
name: Rule name.
|
||||
srcs: Source .fbs files. Sent in order to the compiler.
|
||||
srcs_filegroup_name: Name of the output filegroup that holds srcs. Pass this
|
||||
filegroup into the `includes` parameter of any other
|
||||
flatbuffer_cc_library that depends on this one's schemas.
|
||||
out_prefix: Prepend this path to the front of all generated files. Usually
|
||||
is a directory name.
|
||||
includes: Optional, list of filegroups of schemas that the srcs depend on.
|
||||
** SEE REMARKS BELOW **
|
||||
include_paths: Optional, list of paths the includes files can be found in.
|
||||
compatible_with: Optional, passed to genrule for environments this rule
|
||||
can be built for
|
||||
flatc_args: Optional list of additional arguments to pass to flatc
|
||||
(e.g. --gen-mutable).
|
||||
visibility: The visibility of the generated cc_library. By default, use the
|
||||
default visibility of the project.
|
||||
srcs_filegroup_visibility: The visibility of the generated srcs filegroup.
|
||||
By default, use the value of the visibility parameter above.
|
||||
gen_reflections: Optional, if true this will generate the flatbuffer
|
||||
reflection binaries for the schemas.
|
||||
'''
|
||||
output_headers = [
|
||||
(out_prefix + "%s_generated.h") % (s.replace(".fbs", "").split("/")[-1])
|
||||
for s in srcs
|
||||
]
|
||||
reflection_name = "%s_reflection" % name if gen_reflections else ""
|
||||
|
||||
flatbuffer_library_public(
|
||||
name = "%s_srcs" % (name),
|
||||
srcs = srcs,
|
||||
outs = output_headers,
|
||||
language_flag = "-c",
|
||||
out_prefix = out_prefix,
|
||||
includes = includes,
|
||||
include_paths = include_paths,
|
||||
compatible_with = compatible_with,
|
||||
flatc_args = flatc_args,
|
||||
reflection_name = reflection_name,
|
||||
reflection_visibility = visibility,
|
||||
)
|
||||
native.cc_library(
|
||||
name = name,
|
||||
hdrs = output_headers,
|
||||
srcs = output_headers,
|
||||
features = [
|
||||
"-parse_headers",
|
||||
],
|
||||
deps = [
|
||||
"@flatbuffers//:runtime_cc",
|
||||
],
|
||||
includes = ["."],
|
||||
linkstatic = 1,
|
||||
visibility = visibility,
|
||||
compatible_with = compatible_with,
|
||||
)
|
||||
|
||||
# A filegroup for the `srcs`. That is, all the schema files for this
|
||||
# Flatbuffer set.
|
||||
native.filegroup(
|
||||
name = srcs_filegroup_name if srcs_filegroup_name else "%s_includes" % (name),
|
||||
srcs = srcs,
|
||||
visibility = srcs_filegroup_visibility if srcs_filegroup_visibility != None else visibility,
|
||||
compatible_with = compatible_with,
|
||||
)
|
||||
|
||||
FlatbufferInfo = provider(
|
||||
"Custom provider to track dependencies transitively.",
|
||||
fields = {
|
||||
"transitive_srcs": "flatbuffer schema definitions.",
|
||||
},
|
||||
)
|
||||
|
||||
def _flatbuffer_schemas_aspect_impl(target, ctx):
|
||||
_ignore = [target]
|
||||
transitive_srcs = depset()
|
||||
if hasattr(ctx.rule.attr, "deps"):
|
||||
for dep in ctx.rule.attr.deps:
|
||||
if FlatbufferInfo in dep:
|
||||
transitive_srcs = depset(dep[FlatbufferInfo].transitive_srcs, transitive = [transitive_srcs]) # buildifier: disable=overly-nested-depset
|
||||
if hasattr(ctx.rule.attr, "srcs"):
|
||||
for src in ctx.rule.attr.srcs:
|
||||
if FlatbufferInfo in src:
|
||||
transitive_srcs = depset(src[FlatbufferInfo].transitive_srcs, transitive = [transitive_srcs]) # buildifier: disable=overly-nested-depset
|
||||
for f in src.files:
|
||||
if f.extension == "fbs":
|
||||
transitive_srcs = depset([f], transitive = [transitive_srcs]) # buildifier: disable=overly-nested-depset
|
||||
return [FlatbufferInfo(transitive_srcs = transitive_srcs)]
|
||||
|
||||
# An aspect that runs over all dependencies and transitively collects
|
||||
# flatbuffer schema files.
|
||||
_flatbuffer_schemas_aspect = aspect(
|
||||
attr_aspects = [
|
||||
"deps",
|
||||
"srcs",
|
||||
],
|
||||
implementation = _flatbuffer_schemas_aspect_impl,
|
||||
)
|
||||
|
||||
# Rule to invoke the flatbuffer compiler.
|
||||
def _gen_flatbuffer_srcs_impl(ctx):
|
||||
outputs = ctx.attr.outputs
|
||||
include_paths = ctx.attr.include_paths
|
||||
if ctx.attr.no_includes:
|
||||
no_includes_statement = ["--no-includes"]
|
||||
else:
|
||||
no_includes_statement = []
|
||||
|
||||
if ctx.attr.language_flag == "--python":
|
||||
onefile_statement = ["--gen-onefile"]
|
||||
else:
|
||||
onefile_statement = []
|
||||
|
||||
# Need to generate all files in a directory.
|
||||
if not outputs:
|
||||
outputs = [ctx.actions.declare_directory("{}_all".format(ctx.attr.name))]
|
||||
output_directory = outputs[0].path
|
||||
else:
|
||||
outputs = [ctx.actions.declare_file(output) for output in outputs]
|
||||
output_directory = outputs[0].dirname
|
||||
|
||||
deps = depset(ctx.files.srcs + ctx.files.deps, transitive = [
|
||||
dep[FlatbufferInfo].transitive_srcs
|
||||
for dep in ctx.attr.deps
|
||||
if FlatbufferInfo in dep
|
||||
])
|
||||
|
||||
include_paths_cmd_line = []
|
||||
for s in include_paths:
|
||||
include_paths_cmd_line.extend(["-I", s])
|
||||
|
||||
for src in ctx.files.srcs:
|
||||
ctx.actions.run(
|
||||
inputs = deps,
|
||||
outputs = outputs,
|
||||
executable = ctx.executable._flatc,
|
||||
arguments = [
|
||||
ctx.attr.language_flag,
|
||||
"-o",
|
||||
output_directory,
|
||||
# Allow for absolute imports and referencing of generated files.
|
||||
"-I",
|
||||
"./",
|
||||
"-I",
|
||||
ctx.genfiles_dir.path,
|
||||
"-I",
|
||||
ctx.bin_dir.path,
|
||||
] + no_includes_statement +
|
||||
onefile_statement +
|
||||
include_paths_cmd_line + [
|
||||
"--no-union-value-namespacing",
|
||||
"--gen-object-api",
|
||||
src.path,
|
||||
],
|
||||
progress_message = "Generating flatbuffer files for {}:".format(src),
|
||||
)
|
||||
return [
|
||||
DefaultInfo(files = depset(outputs)),
|
||||
]
|
||||
|
||||
_gen_flatbuffer_srcs = rule(
|
||||
_gen_flatbuffer_srcs_impl,
|
||||
attrs = {
|
||||
"srcs": attr.label_list(
|
||||
allow_files = [".fbs"],
|
||||
mandatory = True,
|
||||
),
|
||||
"outputs": attr.string_list(
|
||||
default = [],
|
||||
mandatory = False,
|
||||
),
|
||||
"deps": attr.label_list(
|
||||
default = [],
|
||||
mandatory = False,
|
||||
aspects = [_flatbuffer_schemas_aspect],
|
||||
),
|
||||
"include_paths": attr.string_list(
|
||||
default = [],
|
||||
mandatory = False,
|
||||
),
|
||||
"language_flag": attr.string(
|
||||
mandatory = True,
|
||||
),
|
||||
"no_includes": attr.bool(
|
||||
default = False,
|
||||
mandatory = False,
|
||||
),
|
||||
"_flatc": attr.label(
|
||||
default = Label("@flatbuffers//:flatc"),
|
||||
executable = True,
|
||||
cfg = "host",
|
||||
),
|
||||
},
|
||||
output_to_genfiles = True,
|
||||
)
|
||||
|
||||
def flatbuffer_py_strip_prefix_srcs(name, srcs = [], strip_prefix = ""):
|
||||
"""Strips path prefix.
|
||||
|
||||
Args:
|
||||
name: Rule name. (required)
|
||||
srcs: Source .py files. (required)
|
||||
strip_prefix: Path that needs to be stripped from the srcs filepaths. (required)
|
||||
"""
|
||||
for src in srcs:
|
||||
native.genrule(
|
||||
name = name + "_" + src.replace(".", "_").replace("/", "_"),
|
||||
srcs = [src],
|
||||
outs = [src.replace(strip_prefix, "")],
|
||||
cmd = "cp $< $@",
|
||||
)
|
||||
|
||||
def _concat_flatbuffer_py_srcs_impl(ctx):
|
||||
# Merge all generated python files. The files are concatenated and the
|
||||
# import statements are removed. Finally we import the flatbuffer runtime
|
||||
# library.
|
||||
# IMPORTANT: Our Windows shell does not support "find ... -exec" properly.
|
||||
# If changing the commandline here, please build wheels and run smoke tests
|
||||
# on all the three operation systems.
|
||||
command = "echo 'import flatbuffers\n' > %s; "
|
||||
command += "for f in $(find %s -name '*.py'); do cat $f | sed '/import flatbuffers/d' >> %s; done "
|
||||
ctx.actions.run_shell(
|
||||
inputs = ctx.attr.deps[0].files,
|
||||
outputs = [ctx.outputs.out],
|
||||
command = command % (
|
||||
ctx.outputs.out.path,
|
||||
ctx.attr.deps[0].files.to_list()[0].path,
|
||||
ctx.outputs.out.path,
|
||||
),
|
||||
)
|
||||
|
||||
_concat_flatbuffer_py_srcs = rule(
|
||||
_concat_flatbuffer_py_srcs_impl,
|
||||
attrs = {
|
||||
"deps": attr.label_list(mandatory = True),
|
||||
},
|
||||
output_to_genfiles = True,
|
||||
outputs = {"out": "%{name}.py"},
|
||||
)
|
||||
|
||||
def flatbuffer_py_library(
|
||||
name,
|
||||
srcs,
|
||||
deps = [],
|
||||
include_paths = []):
|
||||
"""A py_library with the generated reader/writers for the given schema.
|
||||
|
||||
This rule assumes that the schema files define non-conflicting names, so that
|
||||
they can be merged in a single file. This is e.g. the case if only a single
|
||||
namespace is used.
|
||||
The rule call the flatbuffer compiler for all schema files and merges the
|
||||
generated python files into a single file that is wrapped in a py_library.
|
||||
|
||||
Args:
|
||||
name: Rule name. (required)
|
||||
srcs: List of source .fbs files. (required)
|
||||
deps: List of dependencies.
|
||||
include_paths: Optional, list of paths the includes files can be found in.
|
||||
"""
|
||||
all_srcs = "{}_srcs".format(name)
|
||||
_gen_flatbuffer_srcs(
|
||||
name = all_srcs,
|
||||
srcs = srcs,
|
||||
language_flag = "--python",
|
||||
deps = deps,
|
||||
include_paths = include_paths,
|
||||
)
|
||||
all_srcs_no_include = "{}_srcs_no_include".format(name)
|
||||
_gen_flatbuffer_srcs(
|
||||
name = all_srcs_no_include,
|
||||
srcs = srcs,
|
||||
language_flag = "--python",
|
||||
deps = deps,
|
||||
no_includes = True,
|
||||
include_paths = include_paths,
|
||||
)
|
||||
|
||||
# TODO: Remove the concatnation rule with 2.0.6 update.
|
||||
concat_py_srcs = "{}_generated".format(name)
|
||||
_concat_flatbuffer_py_srcs(
|
||||
name = concat_py_srcs,
|
||||
deps = [
|
||||
":{}".format(all_srcs_no_include),
|
||||
],
|
||||
)
|
||||
native.py_library(
|
||||
name = name,
|
||||
srcs = [
|
||||
":{}".format(concat_py_srcs),
|
||||
],
|
||||
srcs_version = "PY2AND3",
|
||||
deps = deps,
|
||||
)
|
||||
|
||||
def flatbuffer_java_library(
|
||||
name,
|
||||
srcs,
|
||||
custom_package = "",
|
||||
package_prefix = "",
|
||||
include_paths = DEFAULT_INCLUDE_PATHS,
|
||||
flatc_args = DEFAULT_FLATC_ARGS,
|
||||
visibility = None):
|
||||
"""A java library with the generated reader/writers for the given flatbuffer definitions.
|
||||
|
||||
Args:
|
||||
name: Rule name. (required)
|
||||
srcs: List of source .fbs files including all includes. (required)
|
||||
custom_package: Package name of generated Java files. If not specified
|
||||
namespace in the schema files will be used. (optional)
|
||||
package_prefix: like custom_package, but prefixes to the existing
|
||||
namespace. (optional)
|
||||
include_paths: List of paths that includes files can be found in. (optional)
|
||||
flatc_args: List of additional arguments to pass to flatc. (optional)
|
||||
visibility: Visibility setting for the java_library rule. (optional)
|
||||
"""
|
||||
out_srcjar = "java_%s_all.srcjar" % name
|
||||
flatbuffer_java_srcjar(
|
||||
name = "%s_srcjar" % name,
|
||||
srcs = srcs,
|
||||
out = out_srcjar,
|
||||
custom_package = custom_package,
|
||||
flatc_args = flatc_args,
|
||||
include_paths = include_paths,
|
||||
package_prefix = package_prefix,
|
||||
)
|
||||
|
||||
native.filegroup(
|
||||
name = "%s.srcjar" % name,
|
||||
srcs = [out_srcjar],
|
||||
)
|
||||
|
||||
native.java_library(
|
||||
name = name,
|
||||
srcs = [out_srcjar],
|
||||
javacopts = ["-source 7 -target 7"],
|
||||
deps = [
|
||||
"@flatbuffers//:runtime_java",
|
||||
],
|
||||
visibility = visibility,
|
||||
)
|
||||
|
||||
def flatbuffer_java_srcjar(
|
||||
name,
|
||||
srcs,
|
||||
out,
|
||||
custom_package = "",
|
||||
package_prefix = "",
|
||||
include_paths = DEFAULT_INCLUDE_PATHS,
|
||||
flatc_args = DEFAULT_FLATC_ARGS):
|
||||
"""Generate flatbuffer Java source files.
|
||||
|
||||
Args:
|
||||
name: Rule name. (required)
|
||||
srcs: List of source .fbs files including all includes. (required)
|
||||
out: Output file name. (required)
|
||||
custom_package: Package name of generated Java files. If not specified
|
||||
namespace in the schema files will be used. (optional)
|
||||
package_prefix: like custom_package, but prefixes to the existing
|
||||
namespace. (optional)
|
||||
include_paths: List of paths that includes files can be found in. (optional)
|
||||
flatc_args: List of additional arguments to pass to flatc. (optional)
|
||||
"""
|
||||
command_fmt = """set -e
|
||||
tmpdir=$(@D)
|
||||
schemas=$$tmpdir/schemas
|
||||
java_root=$$tmpdir/java
|
||||
rm -rf $$schemas
|
||||
rm -rf $$java_root
|
||||
mkdir -p $$schemas
|
||||
mkdir -p $$java_root
|
||||
|
||||
for src in $(SRCS); do
|
||||
dest=$$schemas/$$src
|
||||
rm -rf $$(dirname $$dest)
|
||||
mkdir -p $$(dirname $$dest)
|
||||
if [ -z "{custom_package}" ] && [ -z "{package_prefix}" ]; then
|
||||
cp -f $$src $$dest
|
||||
else
|
||||
if [ -z "{package_prefix}" ]; then
|
||||
sed -e "s/namespace\\s.*/namespace {custom_package};/" $$src > $$dest
|
||||
else
|
||||
sed -e "s/namespace \\([^;]\\+\\);/namespace {package_prefix}.\\1;/" $$src > $$dest
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
flatc_arg_I="-I $$tmpdir/schemas"
|
||||
for include_path in {include_paths}; do
|
||||
flatc_arg_I="$$flatc_arg_I -I $$schemas/$$include_path"
|
||||
done
|
||||
|
||||
flatc_additional_args=
|
||||
for arg in {flatc_args}; do
|
||||
flatc_additional_args="$$flatc_additional_args $$arg"
|
||||
done
|
||||
|
||||
for src in $(SRCS); do
|
||||
$(location {flatc_path}) $$flatc_arg_I --java $$flatc_additional_args -o $$java_root $$schemas/$$src
|
||||
done
|
||||
|
||||
$(location {zip_files}) -export_zip_path=$@ -file_directory=$$java_root
|
||||
"""
|
||||
genrule_cmd = command_fmt.format(
|
||||
package_name = native.package_name(),
|
||||
custom_package = custom_package,
|
||||
package_prefix = package_prefix,
|
||||
flatc_path = flatc_path,
|
||||
zip_files = zip_files,
|
||||
include_paths = " ".join(include_paths),
|
||||
flatc_args = " ".join(flatc_args),
|
||||
)
|
||||
|
||||
native.genrule(
|
||||
name = name,
|
||||
srcs = srcs,
|
||||
outs = [out],
|
||||
tools = [flatc_path, zip_files],
|
||||
cmd = genrule_cmd,
|
||||
)
|
||||
|
||||
def flatbuffer_android_library(
|
||||
name,
|
||||
srcs,
|
||||
custom_package = "",
|
||||
package_prefix = "",
|
||||
include_paths = DEFAULT_INCLUDE_PATHS,
|
||||
flatc_args = DEFAULT_FLATC_ARGS,
|
||||
visibility = None):
|
||||
"""An android_library with the generated reader/writers for the given flatbuffer definitions.
|
||||
|
||||
Args:
|
||||
name: Rule name. (required)
|
||||
srcs: List of source .fbs files including all includes. (required)
|
||||
custom_package: Package name of generated Java files. If not specified
|
||||
namespace in the schema files will be used. (optional)
|
||||
package_prefix: like custom_package, but prefixes to the existing
|
||||
namespace. (optional)
|
||||
include_paths: List of paths that includes files can be found in. (optional)
|
||||
flatc_args: List of additional arguments to pass to flatc. (optional)
|
||||
visibility: Visibility setting for the android_library rule. (optional)
|
||||
"""
|
||||
out_srcjar = "android_%s_all.srcjar" % name
|
||||
flatbuffer_java_srcjar(
|
||||
name = "%s_srcjar" % name,
|
||||
srcs = srcs,
|
||||
out = out_srcjar,
|
||||
custom_package = custom_package,
|
||||
flatc_args = flatc_args,
|
||||
include_paths = include_paths,
|
||||
package_prefix = package_prefix,
|
||||
)
|
||||
|
||||
native.filegroup(
|
||||
name = "%s.srcjar" % name,
|
||||
srcs = [out_srcjar],
|
||||
)
|
||||
|
||||
# To support org.checkerframework.dataflow.qual.Pure.
|
||||
checkerframework_annotations = [
|
||||
"@org_checkerframework_qual",
|
||||
] if "--java-checkerframework" in flatc_args else []
|
||||
|
||||
android_library(
|
||||
name = name,
|
||||
srcs = [out_srcjar],
|
||||
javacopts = ["-source 7 -target 7"],
|
||||
visibility = visibility,
|
||||
deps = [
|
||||
"@flatbuffers//:runtime_android",
|
||||
] + checkerframework_annotations,
|
||||
)
|
||||
+19
@@ -0,0 +1,19 @@
|
||||
"""Loads the Flatbuffers library, used by TF Lite."""
|
||||
|
||||
load("//third_party:repo.bzl", "third_party_http_archive")
|
||||
|
||||
def repo():
|
||||
third_party_http_archive(
|
||||
name = "flatbuffers",
|
||||
strip_prefix = "flatbuffers-2.0.6",
|
||||
sha256 = "e2dc24985a85b278dd06313481a9ca051d048f9474e0f199e372fea3ea4248c9",
|
||||
urls = [
|
||||
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/google/flatbuffers/archive/v2.0.6.tar.gz",
|
||||
"https://github.com/google/flatbuffers/archive/v2.0.6.tar.gz",
|
||||
],
|
||||
build_file = "//third_party/flatbuffers:BUILD.bazel",
|
||||
delete = ["build_defs.bzl", "BUILD.bazel"],
|
||||
link_files = {
|
||||
"//third_party/flatbuffers:build_defs.bzl": "build_defs.bzl",
|
||||
},
|
||||
)
|
||||
+8
-18
@@ -1634,7 +1634,7 @@ new file mode 100644
|
||||
index 00000000000..2723216f324
|
||||
--- /dev/null
|
||||
+++ b/tensorflow/lite/delegates/gpu/common/tasks/mediapipe/transform_tensor_bilinear.cc
|
||||
@@ -0,0 +1,133 @@
|
||||
@@ -0,0 +1,123 @@
|
||||
+#include "tensorflow/lite/delegates/gpu/common/tasks/mediapipe/transform_tensor_bilinear.h"
|
||||
+
|
||||
+#include <string>
|
||||
@@ -1707,22 +1707,14 @@ index 00000000000..2723216f324
|
||||
+ c += " float4 p1 = INIT_FLOAT4(0.0f);\n";
|
||||
+ c += " float4 p2 = INIT_FLOAT4(0.0f);\n";
|
||||
+ c += " float4 p3 = INIT_FLOAT4(0.0f);\n";
|
||||
+ const auto src_tensor_type = op_def.src_tensors[0].storage_type;
|
||||
+ const bool buffer_type = src_tensor_type == TensorStorageType::BUFFER ||
|
||||
+ src_tensor_type == TensorStorageType::IMAGE_BUFFER;
|
||||
+ auto read_src = [&](const std::string& result, const std::string& xc,
|
||||
+ const std::string& yc, const std::string& zc) {
|
||||
+ if (buffer_type) {
|
||||
+ c += " if(" + xc + " >= 0 && " + yc + " >= 0 && " + xc +
|
||||
+ " < args.src_tensor.Width() && " + yc +
|
||||
+ " < args.src_tensor.Height()) {\n";
|
||||
+ c += " " + result + " = args.src_tensor.Read<float>(" + xc + ", " +
|
||||
+ yc + ", " + zc + ");\n";
|
||||
+ c += " }\n";
|
||||
+ } else {
|
||||
+ c += " " + result + " = args.src_tensor.Read<float>(" + xc + ", " +
|
||||
+ yc + ", " + zc + ");\n";
|
||||
+ }
|
||||
+ c += " if(" + xc + " >= 0 && " + yc + " >= 0 && " + xc +
|
||||
+ " < args.src_tensor.Width() && " + yc +
|
||||
+ " < args.src_tensor.Height()) {\n";
|
||||
+ c += " " + result + " = args.src_tensor.Read<float>(" + xc + ", " +
|
||||
+ yc + ", " + zc + ");\n";
|
||||
+ c += " }\n";
|
||||
+ };
|
||||
+ read_src("p0", "st.x", "st.y", "Z");
|
||||
+ read_src("p1", "st.z", "st.y", "Z");
|
||||
@@ -1755,9 +1747,7 @@ index 00000000000..2723216f324
|
||||
+ const OperationDef& definition,
|
||||
+ const TransformTensorBilinearAttributes& attr) {
|
||||
+ GPUOperation op(definition);
|
||||
+ auto src_desc = definition.src_tensors[0];
|
||||
+ src_desc.SetAddressMode(AddressMode::kZero);
|
||||
+ op.AddSrcTensor("src_tensor", src_desc);
|
||||
+ op.AddSrcTensor("src_tensor", definition.src_tensors[0]);
|
||||
+ op.AddSrcTensor("matrix_transform", definition.src_tensors[1]);
|
||||
+ op.AddDstTensor("dst_tensor", definition.dst_tensors[0]);
|
||||
+ op.code_ =
|
||||
|
||||
Vendored
+152
@@ -0,0 +1,152 @@
|
||||
# Copyright 2022 The MediaPipe Authors. All rights reserved.
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
"""Utilities for defining TensorFlow Lite Support Bazel dependencies."""
|
||||
|
||||
_SINGLE_URL_WHITELIST = []
|
||||
|
||||
def _is_windows(ctx):
|
||||
return ctx.os.name.lower().find("windows") != -1
|
||||
|
||||
def _wrap_bash_cmd(ctx, cmd):
|
||||
if _is_windows(ctx):
|
||||
bazel_sh = _get_env_var(ctx, "BAZEL_SH")
|
||||
if not bazel_sh:
|
||||
fail("BAZEL_SH environment variable is not set")
|
||||
cmd = [bazel_sh, "-l", "-c", " ".join(["\"%s\"" % s for s in cmd])]
|
||||
return cmd
|
||||
|
||||
def _get_env_var(ctx, name):
|
||||
if name in ctx.os.environ:
|
||||
return ctx.os.environ[name]
|
||||
else:
|
||||
return None
|
||||
|
||||
# Checks if we should use the system lib instead of the bundled one
|
||||
def _use_system_lib(ctx, name):
|
||||
syslibenv = _get_env_var(ctx, "TF_SYSTEM_LIBS")
|
||||
if syslibenv:
|
||||
for n in syslibenv.strip().split(","):
|
||||
if n.strip() == name:
|
||||
return True
|
||||
return False
|
||||
|
||||
# Executes specified command with arguments and calls 'fail' if it exited with
|
||||
# non-zero code
|
||||
def _execute_and_check_ret_code(repo_ctx, cmd_and_args):
|
||||
result = repo_ctx.execute(cmd_and_args, timeout = 60)
|
||||
if result.return_code != 0:
|
||||
fail(("Non-zero return code({1}) when executing '{0}':\n" + "Stdout: {2}\n" +
|
||||
"Stderr: {3}").format(
|
||||
" ".join([str(x) for x in cmd_and_args]),
|
||||
result.return_code,
|
||||
result.stdout,
|
||||
result.stderr,
|
||||
))
|
||||
|
||||
# Apply a patch_file to the repository root directory
|
||||
# Runs 'patch -p1' on both Windows and Unix.
|
||||
def _apply_patch(ctx, patch_file):
|
||||
patch_command = ["patch", "-p1", "-d", ctx.path("."), "-i", ctx.path(patch_file)]
|
||||
cmd = _wrap_bash_cmd(ctx, patch_command)
|
||||
_execute_and_check_ret_code(ctx, cmd)
|
||||
|
||||
def _apply_delete(ctx, paths):
|
||||
for path in paths:
|
||||
if path.startswith("/"):
|
||||
fail("refusing to rm -rf path starting with '/': " + path)
|
||||
if ".." in path:
|
||||
fail("refusing to rm -rf path containing '..': " + path)
|
||||
cmd = _wrap_bash_cmd(ctx, ["rm", "-rf"] + [ctx.path(path) for path in paths])
|
||||
_execute_and_check_ret_code(ctx, cmd)
|
||||
|
||||
def _third_party_http_archive(ctx):
|
||||
"""Downloads and creates Bazel repos for dependencies.
|
||||
|
||||
This is a swappable replacement for both http_archive() and
|
||||
new_http_archive() that offers some additional features. It also helps
|
||||
ensure best practices are followed.
|
||||
"""
|
||||
if ("mirror.tensorflow.org" not in ctx.attr.urls[0] and
|
||||
(len(ctx.attr.urls) < 2 and
|
||||
ctx.attr.name not in _SINGLE_URL_WHITELIST.to_list())):
|
||||
fail("third_party_http_archive(urls) must have redundant URLs. The " +
|
||||
"mirror.tensorflow.org URL must be present and it must come first. " +
|
||||
"Even if you don't have permission to mirror the file, please " +
|
||||
"put the correctly formatted mirror URL there anyway, because " +
|
||||
"someone will come along shortly thereafter and mirror the file.")
|
||||
|
||||
use_syslib = _use_system_lib(ctx, ctx.attr.name)
|
||||
|
||||
# Use "BUILD.bazel" to avoid conflict with third party projects that contain a
|
||||
# file or directory called "BUILD"
|
||||
buildfile_path = ctx.path("BUILD.bazel")
|
||||
|
||||
if use_syslib:
|
||||
if ctx.attr.system_build_file == None:
|
||||
fail("Bazel was configured with TF_SYSTEM_LIBS to use a system " +
|
||||
"library for %s, but no system build file for %s was configured. " +
|
||||
"Please add a system_build_file attribute to the repository rule" +
|
||||
"for %s." % (ctx.attr.name, ctx.attr.name, ctx.attr.name))
|
||||
ctx.symlink(Label(ctx.attr.system_build_file), buildfile_path)
|
||||
|
||||
else:
|
||||
ctx.download_and_extract(
|
||||
ctx.attr.urls,
|
||||
"",
|
||||
ctx.attr.sha256,
|
||||
ctx.attr.type,
|
||||
ctx.attr.strip_prefix,
|
||||
)
|
||||
if ctx.attr.delete:
|
||||
_apply_delete(ctx, ctx.attr.delete)
|
||||
if ctx.attr.patch_file != None:
|
||||
_apply_patch(ctx, ctx.attr.patch_file)
|
||||
ctx.symlink(Label(ctx.attr.build_file), buildfile_path)
|
||||
|
||||
link_dict = {}
|
||||
if use_syslib:
|
||||
link_dict.update(ctx.attr.system_link_files)
|
||||
|
||||
for internal_src, external_dest in ctx.attr.link_files.items():
|
||||
# if syslib and link exists in both, use the system one
|
||||
if external_dest not in link_dict.values():
|
||||
link_dict[internal_src] = external_dest
|
||||
|
||||
for internal_src, external_dest in link_dict.items():
|
||||
ctx.symlink(Label(internal_src), ctx.path(external_dest))
|
||||
|
||||
# For link_files, specify each dict entry as:
|
||||
# "//path/to/source:file": "localfile"
|
||||
third_party_http_archive = repository_rule(
|
||||
attrs = {
|
||||
"sha256": attr.string(mandatory = True),
|
||||
"urls": attr.string_list(
|
||||
mandatory = True,
|
||||
allow_empty = False,
|
||||
),
|
||||
"strip_prefix": attr.string(),
|
||||
"type": attr.string(),
|
||||
"delete": attr.string_list(),
|
||||
"build_file": attr.string(mandatory = True),
|
||||
"system_build_file": attr.string(mandatory = False),
|
||||
"patch_file": attr.label(),
|
||||
"link_files": attr.string_dict(),
|
||||
"system_link_files": attr.string_dict(),
|
||||
},
|
||||
environ = [
|
||||
"TF_SYSTEM_LIBS",
|
||||
],
|
||||
implementation = _third_party_http_archive,
|
||||
)
|
||||
+38
@@ -0,0 +1,38 @@
|
||||
This diff patches commit a0f49e63, which has been released os part of Tensorflow
|
||||
Text 2.9. We can remove this patch once we update our dependency. This has not
|
||||
been done yet due to a Protobuf incompatibility introduced by this update.
|
||||
|
||||
diff --git a/tensorflow_text/core/kernels/regex_split.cc b/tensorflow_text/core/kernels/regex_split.cc
|
||||
index 473b922e..db422781 100644
|
||||
--- a/tensorflow_text/core/kernels/regex_split.cc
|
||||
+++ b/tensorflow_text/core/kernels/regex_split.cc
|
||||
@@ -45,24 +45,24 @@ void RegexSplitImpl(absl::string_view input, const RE2& re2,
|
||||
tokens->push_back(token);
|
||||
// Mark the end of the last token
|
||||
begin_offsets->push_back(token.data() - input.data());
|
||||
- end_offsets->push_back(token.data() + token.length() - input.begin());
|
||||
+ end_offsets->push_back(token.data() + token.length() - input.data());
|
||||
}
|
||||
|
||||
if (should_include_delim) {
|
||||
// If desired, include the deliminator as a token.
|
||||
tokens->push_back(extracted_delim_token);
|
||||
// Mark the end of the token at the end of the beginning of the delimiter.
|
||||
- begin_offsets->push_back(extracted_delim_token.data() - input.begin());
|
||||
+ begin_offsets->push_back(extracted_delim_token.data() - input.data());
|
||||
end_offsets->push_back(extracted_delim_token.data() +
|
||||
- extracted_delim_token.length() - input.begin());
|
||||
+ extracted_delim_token.length() - input.data());
|
||||
}
|
||||
}
|
||||
|
||||
// Close the last token.
|
||||
if (!leftover.empty()) {
|
||||
tokens->push_back(leftover);
|
||||
- begin_offsets->push_back(leftover.data() - input.begin());
|
||||
- end_offsets->push_back(leftover.data() + leftover.length() - input.begin());
|
||||
+ begin_offsets->push_back(leftover.data() - input.data());
|
||||
+ end_offsets->push_back(leftover.data() + leftover.length() - input.data());
|
||||
}
|
||||
}
|
||||
|
||||
+33
@@ -0,0 +1,33 @@
|
||||
diff --git a/tensorflow_text/core/kernels/BUILD b/tensorflow_text/core/kernels/BUILD
|
||||
index bdca365..1c20eae 100644
|
||||
--- a/tensorflow_text/core/kernels/BUILD
|
||||
+++ b/tensorflow_text/core/kernels/BUILD
|
||||
@@ -209,8 +209,12 @@ cc_library(
|
||||
name = "regex_split",
|
||||
srcs = ["regex_split.cc"],
|
||||
hdrs = ["regex_split.h"],
|
||||
- deps = OSS_DEPS + [
|
||||
+ deps = [
|
||||
# absl/strings dep
|
||||
+ "@com_google_absl//absl/container:inlined_vector",
|
||||
+ "@com_google_absl//absl/strings",
|
||||
+ "@com_google_absl//absl/types:optional",
|
||||
+ "@com_google_absl//absl/types:span",
|
||||
"@com_google_re2//:re2",
|
||||
],
|
||||
)
|
||||
@@ -437,8 +441,12 @@ cc_library(
|
||||
name = "wordpiece_tokenizer",
|
||||
srcs = ["wordpiece_tokenizer.cc"],
|
||||
hdrs = ["wordpiece_tokenizer.h"],
|
||||
- deps = OSS_DEPS + [
|
||||
+ deps = [
|
||||
# absl/strings dep
|
||||
+ "@com_google_absl//absl/container:inlined_vector",
|
||||
+ "@com_google_absl//absl/strings",
|
||||
+ "@com_google_absl//absl/types:optional",
|
||||
+ "@com_google_absl//absl/types:span",
|
||||
"@icu//:common",
|
||||
],
|
||||
)
|
||||
|
||||
Vendored
+78
@@ -0,0 +1,78 @@
|
||||
package(default_visibility = ["//visibility:public"])
|
||||
|
||||
licenses(["notice"])
|
||||
|
||||
COPTS = select({
|
||||
# Don't pass any of the default flags as the VS Compiler does not support
|
||||
# these settings.
|
||||
"@platforms//os:windows": [],
|
||||
"//conditions:default": [
|
||||
"-Wno-dangling-else",
|
||||
"-Wno-format",
|
||||
"-Wno-implicit-function-declaration",
|
||||
"-Wno-incompatible-pointer-types",
|
||||
"-Wno-incompatible-pointer-types-discards-qualifiers",
|
||||
"-Wno-parentheses",
|
||||
"-DIOAPI_NO_64",
|
||||
],
|
||||
})
|
||||
|
||||
cc_library(
|
||||
name = "zlib",
|
||||
srcs = [
|
||||
"adler32.c",
|
||||
"compress.c",
|
||||
"crc32.c",
|
||||
"crc32.h",
|
||||
"deflate.c",
|
||||
"deflate.h",
|
||||
"gzclose.c",
|
||||
"gzguts.h",
|
||||
"gzlib.c",
|
||||
"gzread.c",
|
||||
"gzwrite.c",
|
||||
"infback.c",
|
||||
"inffast.c",
|
||||
"inffast.h",
|
||||
"inffixed.h",
|
||||
"inflate.c",
|
||||
"inflate.h",
|
||||
"inftrees.c",
|
||||
"inftrees.h",
|
||||
"trees.c",
|
||||
"trees.h",
|
||||
"uncompr.c",
|
||||
"zutil.c",
|
||||
"zutil.h",
|
||||
],
|
||||
hdrs = [
|
||||
"zconf.h",
|
||||
"zlib.h",
|
||||
],
|
||||
copts = COPTS,
|
||||
includes = ["."],
|
||||
)
|
||||
|
||||
cc_library(
|
||||
name = "zlib_minizip",
|
||||
srcs = [
|
||||
"contrib/minizip/ioapi.c",
|
||||
"contrib/minizip/unzip.c",
|
||||
"contrib/minizip/zip.c",
|
||||
] + select({
|
||||
"@platforms//os:windows": [
|
||||
"contrib/minizip/iowin32.c",
|
||||
"contrib/minizip/iowin32.h",
|
||||
],
|
||||
"//conditions:default": [],
|
||||
}),
|
||||
hdrs = [
|
||||
"contrib/minizip/crypt.h",
|
||||
"contrib/minizip/ioapi.h",
|
||||
"contrib/minizip/mztools.h",
|
||||
"contrib/minizip/unzip.h",
|
||||
"contrib/minizip/zip.h",
|
||||
],
|
||||
copts = COPTS,
|
||||
deps = [":zlib"],
|
||||
)
|
||||
Vendored
+62
@@ -0,0 +1,62 @@
|
||||
diff -ruN a/contrib/minizip/ioapi.h b/contrib/minizip/ioapi.h
|
||||
--- a/contrib/minizip/ioapi.h
|
||||
+++ b/contrib/minizip/ioapi.h
|
||||
@@ -21,7 +21,7 @@
|
||||
#ifndef _ZLIBIOAPI64_H
|
||||
#define _ZLIBIOAPI64_H
|
||||
|
||||
-#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__))
|
||||
+#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__)) && (!defined(__ANDROID__))
|
||||
|
||||
// Linux needs this to support file operation on files larger then 4+GB
|
||||
// But might need better if/def to select just the platforms that needs them.
|
||||
diff -ruN a/contrib/minizip/miniunz.c b/contrib/minizip/miniunz.c
|
||||
--- a/contrib/minizip/miniunz.c
|
||||
+++ b/contrib/minizip/miniunz.c
|
||||
@@ -12,7 +12,7 @@
|
||||
Copyright (C) 2009-2010 Mathias Svensson ( http://result42.com )
|
||||
*/
|
||||
|
||||
-#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__))
|
||||
+#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__)) && (!defined(__ANDROID__))
|
||||
#ifndef __USE_FILE_OFFSET64
|
||||
#define __USE_FILE_OFFSET64
|
||||
#endif
|
||||
@@ -27,7 +27,7 @@
|
||||
#endif
|
||||
#endif
|
||||
|
||||
-#ifdef __APPLE__
|
||||
+#if defined(__APPLE__) || defined(IOAPI_NO_64)
|
||||
// In darwin and perhaps other BSD variants off_t is a 64 bit value, hence no need for specific 64 bit functions
|
||||
#define FOPEN_FUNC(filename, mode) fopen(filename, mode)
|
||||
#define FTELLO_FUNC(stream) ftello(stream)
|
||||
@@ -50,6 +50,7 @@
|
||||
# include <direct.h>
|
||||
# include <io.h>
|
||||
#else
|
||||
+# include <sys/stat.h>
|
||||
# include <unistd.h>
|
||||
# include <utime.h>
|
||||
#endif
|
||||
diff -ruN a/contrib/minizip/minizip.c b/contrib/minizip/minizip.c
|
||||
--- a/contrib/minizip/minizip.c
|
||||
+++ b/contrib/minizip/minizip.c
|
||||
@@ -13,7 +13,7 @@
|
||||
*/
|
||||
|
||||
|
||||
-#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__))
|
||||
+#if (!defined(_WIN32)) && (!defined(WIN32)) && (!defined(__APPLE__)) && (!defined(__ANDROID__))
|
||||
#ifndef __USE_FILE_OFFSET64
|
||||
#define __USE_FILE_OFFSET64
|
||||
#endif
|
||||
@@ -28,7 +28,7 @@
|
||||
#endif
|
||||
#endif
|
||||
|
||||
-#ifdef __APPLE__
|
||||
+#if defined(__APPLE__) || defined(IOAPI_NO_64)
|
||||
// In darwin and perhaps other BSD variants off_t is a 64 bit value, hence no need for specific 64 bit functions
|
||||
#define FOPEN_FUNC(filename, mode) fopen(filename, mode)
|
||||
#define FTELLO_FUNC(stream) ftello(stream)
|
||||
Reference in New Issue
Block a user