1load("//tensorflow:tensorflow.bzl", "pybind_extension") 2 3package( 4 default_visibility = ["//tensorflow:internal"], 5 licenses = ["notice"], 6) 7 8pybind_extension( 9 name = "_pywrap_analyzer_wrapper", 10 srcs = [ 11 "analyzer_wrapper.cc", 12 ], 13 link_in_framework = True, 14 module_name = "_pywrap_analyzer_wrapper", 15 deps = [ 16 ":model_analyzer", 17 "//tensorflow/compiler/mlir/lite:flatbuffer_to_mlir", 18 "@pybind11", 19 ], 20) 21 22cc_library( 23 name = "model_analyzer", 24 srcs = ["model_analyzer.cc"], 25 hdrs = ["model_analyzer.h"], 26 visibility = ["//visibility:public"], 27 deps = [ 28 "//tensorflow/lite:model_builder", 29 "//tensorflow/lite:version", 30 "//tensorflow/lite/core/api:error_reporter", 31 "//tensorflow/lite/schema:schema_fbs", 32 "//tensorflow/lite/schema:schema_utils", 33 "//tensorflow/lite/tools/versioning:gpu_compatibility", 34 "@com_google_absl//absl/strings", 35 ], 36) 37