1load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") 2 3licenses(["notice"]) 4 5package(default_visibility = [":friends"]) 6 7package_group( 8 name = "friends", 9 packages = [ 10 "//learning/brain/mobile/lite/tools/analyzer/...", 11 "//tensorflow/lite/toco/...", 12 ], 13) 14 15cc_library( 16 name = "tf_tfl_flatbuffer_helpers", 17 srcs = ["tf_tfl_flatbuffer_helpers.cc"], 18 hdrs = ["tf_tfl_flatbuffer_helpers.h"], 19 deps = [ 20 "//tensorflow/compiler/mlir/lite:common", 21 "//tensorflow/compiler/mlir/lite:tensorflow_lite", 22 "//tensorflow/compiler/mlir/lite:tf_tfl_passes", 23 "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer", 24 "//tensorflow/compiler/mlir/lite/metrics:error_collector_inst", 25 "//tensorflow/compiler/mlir/tensorflow", 26 "//tensorflow/compiler/mlir/tensorflow:dump_mlir_util", 27 "//tensorflow/compiler/mlir/tensorflow:import_model", 28 "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", 29 "//tensorflow/core:core_cpu_base", 30 "//tensorflow/core:lib", 31 "//tensorflow/core:protos_all_cc", 32 "//tensorflow/lite/toco:model_flags_proto_cc", 33 "//tensorflow/lite/toco:toco_flags_proto_cc", 34 "//tensorflow/lite/toco:types_proto_cc", 35 "//tensorflow/lite/tools/optimize:reduced_precision_support", 36 "//tensorflow/stream_executor/lib", 37 "@llvm-project//llvm:Support", 38 "@llvm-project//mlir:FuncDialect", 39 "@llvm-project//mlir:IR", 40 "@llvm-project//mlir:Pass", 41 "@llvm-project//mlir:Support", 42 "@llvm-project//mlir:Transforms", 43 ], 44) 45 46cc_library( 47 name = "graphdef_to_tfl_flatbuffer", 48 srcs = ["graphdef_to_tfl_flatbuffer.cc"], 49 hdrs = [ 50 "graphdef_to_tfl_flatbuffer.h", 51 ], 52 deps = [ 53 ":tf_tfl_flatbuffer_helpers", 54 "//tensorflow/compiler/mlir/lite:common", 55 "//tensorflow/compiler/mlir/lite:tensorflow_lite", 56 "//tensorflow/compiler/mlir/lite:tf_tfl_passes", 57 "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer", 58 "//tensorflow/compiler/mlir/tensorflow", 59 "//tensorflow/compiler/mlir/tensorflow:import_model", 60 "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", 61 "//tensorflow/core:lib", 62 "//tensorflow/core:protos_all_cc", 63 "//tensorflow/lite/toco:model_flags_proto_cc", 64 "//tensorflow/lite/toco:toco_flags_proto_cc", 65 "//tensorflow/lite/toco:types_proto_cc", 66 "//tensorflow/lite/tools/optimize:reduced_precision_support", 67 "//tensorflow/stream_executor/lib", 68 "@llvm-project//llvm:Support", 69 "@llvm-project//mlir:FuncDialect", 70 "@llvm-project//mlir:IR", 71 "@llvm-project//mlir:Pass", 72 "@llvm-project//mlir:Support", 73 "@llvm-project//mlir:Transforms", 74 ], 75) 76 77cc_library( 78 name = "saved_model_to_tfl_flatbuffer", 79 srcs = ["saved_model_to_tfl_flatbuffer.cc"], 80 hdrs = [ 81 "saved_model_to_tfl_flatbuffer.h", 82 ], 83 deps = [ 84 ":tf_tfl_flatbuffer_helpers", 85 "//tensorflow/cc/saved_model:loader", 86 "//tensorflow/compiler/mlir/lite:common", 87 "//tensorflow/compiler/mlir/lite:tensorflow_lite", 88 "//tensorflow/compiler/mlir/lite:tf_tfl_passes", 89 "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer", 90 "//tensorflow/compiler/mlir/tensorflow", 91 "//tensorflow/compiler/mlir/tensorflow:import_model", 92 "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", 93 "//tensorflow/core:lib", 94 "//tensorflow/core:protos_all_cc", 95 "//tensorflow/lite/toco:model_flags_proto_cc", 96 "//tensorflow/lite/toco:toco_flags_proto_cc", 97 "//tensorflow/lite/toco:types_proto_cc", 98 "//tensorflow/stream_executor/lib", 99 "@com_google_absl//absl/types:span", 100 "@llvm-project//llvm:Support", 101 "@llvm-project//mlir:FuncDialect", 102 "@llvm-project//mlir:IR", 103 "@llvm-project//mlir:Pass", 104 "@llvm-project//mlir:Support", 105 "@llvm-project//mlir:Transforms", 106 ], 107) 108 109cc_library( 110 name = "jax_to_tfl_flatbuffer", 111 srcs = ["jax_to_tfl_flatbuffer.cc"], 112 hdrs = [ 113 "jax_to_tfl_flatbuffer.h", 114 ], 115 deps = [ 116 ":tf_tfl_flatbuffer_helpers", 117 "//tensorflow/compiler/mlir/lite:common", 118 "//tensorflow/compiler/mlir/lite:tensorflow_lite", 119 "//tensorflow/compiler/mlir/lite:tf_tfl_passes", 120 "//tensorflow/compiler/mlir/lite:tf_to_tfl_flatbuffer", 121 "//tensorflow/compiler/mlir/tensorflow", 122 "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", 123 "//tensorflow/compiler/mlir/xla:hlo_to_mlir_hlo", 124 "//tensorflow/compiler/xla/service:hlo_parser", 125 "//tensorflow/compiler/xla/service:hlo_proto_cc", 126 "//tensorflow/core:lib", 127 "//tensorflow/core:protos_all_cc", 128 "//tensorflow/lite/toco:model_flags_proto_cc", 129 "//tensorflow/lite/toco:toco_flags_proto_cc", 130 "//tensorflow/lite/toco:types_proto_cc", 131 "//tensorflow/stream_executor/lib", 132 "@com_google_absl//absl/strings", 133 "@com_google_absl//absl/types:span", 134 "@llvm-project//llvm:Support", 135 "@llvm-project//mlir:FuncDialect", 136 "@llvm-project//mlir:IR", 137 "@llvm-project//mlir:Pass", 138 "@llvm-project//mlir:Support", 139 "@llvm-project//mlir:Transforms", 140 ], 141) 142 143# Smaller version of flatbuffer_translate which only converts flatbuffer to MLIR. 144cc_library( 145 name = "flatbuffer_to_mlir", 146 srcs = [ 147 "flatbuffer_to_mlir.cc", 148 ], 149 hdrs = [ 150 "flatbuffer_to_mlir.h", 151 ], 152 deps = [ 153 "//tensorflow/compiler/mlir/lite:flatbuffer_import", 154 "@llvm-project//llvm:Support", 155 "@llvm-project//mlir:FuncDialect", 156 "@llvm-project//mlir:IR", 157 "@llvm-project//mlir:QuantOps", 158 "@llvm-project//mlir:Support", 159 "@llvm-project//mlir:TranslateLib", 160 ], 161) 162