| # buildifier: disable=same-origin-load |
| load("//tensorflow:tensorflow.bzl", "filegroup") |
| |
| # buildifier: disable=same-origin-load |
| load("//tensorflow:tensorflow.bzl", "get_compatible_with_cloud") |
| load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") |
| load("//tensorflow:tensorflow.bzl", "tf_cc_binary", "tf_cc_test", "tf_native_cc_binary") |
| load( |
| "//third_party/mlir:tblgen.bzl", |
| "gentbl", |
| ) |
| |
| package( |
| default_visibility = [ |
| # TODO(jpienaar): Make the visibility more restrictive. |
| ":friends", |
| "//tensorflow/lite/experimental/tf_runtime:__subpackages__", |
| ], |
| licenses = ["notice"], |
| ) |
| |
| package_group( |
| name = "friends", |
| includes = ["//third_party/mlir:subpackages"], |
| packages = [ |
| "//learning/brain/experimental/mlir/tflite/tfmrt/...", |
| "//learning/brain/mlir/...", |
| "//tensorflow/compiler/mlir/...", |
| "//third_party/iree/...", |
| ], |
| ) |
| |
| filegroup( |
| name = "tensorflow_lite_ops_td_files", |
| srcs = [ |
| "ir/tfl_op_interfaces.td", |
| "ir/tfl_ops.td", |
| "ir/tfl_structs.td", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_td_files", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files", |
| "@llvm-project//mlir:OpBaseTdFiles", |
| "@llvm-project//mlir:SideEffectTdFiles", |
| "@llvm-project//mlir:include/mlir/Interfaces/InferTypeOpInterface.td", |
| "@llvm-project//mlir:include/mlir/Interfaces/LoopLikeInterface.td", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_ops_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-op-decls", |
| "ir/tfl_ops.h.inc", |
| ), |
| ( |
| "-gen-op-defs", |
| "ir/tfl_ops.cc.inc", |
| ), |
| ( |
| "-gen-dialect-doc", |
| "g3doc/tfl_ops.md", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "ir/tfl_ops.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_structs_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-struct-attr-decls", |
| "ir/tfl_structs.h.inc", |
| ), |
| ( |
| "-gen-struct-attr-defs", |
| "ir/tfl_structs.cc.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "ir/tfl_structs.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_op_interfaces_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-op-interface-decls", |
| "ir/tfl_ops_interface.h.inc", |
| ), |
| ( |
| "-gen-op-interface-defs", |
| "ir/tfl_ops_interface.cc.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "ir/tfl_op_interfaces.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_prepare_tf_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_prepare_tf.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/prepare_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_optimize_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_lower_static_tensor_list_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_lower_static_tensor_list.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/tensorlist_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_legalize_tf_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_legalize_tf.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/legalize_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_optimize_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_optimize.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/optimize_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| "utils/utils.td", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_quantize_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_quantize.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/quantize_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| ], |
| ) |
| |
| gentbl( |
| name = "tensorflow_lite_post_quantize_inc_gen", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "-gen-rewriters", |
| "transforms/generated_post_quantize.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "transforms/post_quantize_patterns.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "@llvm-project//mlir:StdOpsTdFiles", |
| ], |
| ) |
| |
| cc_library( |
| name = "validators", |
| srcs = [ |
| "utils/validators.cc", |
| ], |
| hdrs = [ |
| "utils/validators.h", |
| ], |
| deps = [ |
| "@llvm-project//mlir:Dialect", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| ], |
| ) |
| |
| cc_library( |
| name = "tensorflow_lite", |
| srcs = [ |
| "ir/tfl_ops.cc", |
| "ir/tfl_ops.cc.inc", |
| "ir/tfl_ops.h.inc", |
| "ir/tfl_ops_interface.cc.inc", |
| "ir/tfl_ops_interface.h.inc", |
| "ir/tfl_structs.cc.inc", |
| "runtime_verifiers.inc", |
| "utils/attribute_utils.cc", |
| ], |
| hdrs = [ |
| "ir/tfl_ops.h", |
| "ir/tfl_structs.h.inc", |
| "transforms/passes.h", |
| "utils/attribute_utils.h", |
| "@llvm-project//mlir:include/mlir/Transforms/InliningUtils.h", |
| ], |
| deps = [ |
| ":cost_estimators", |
| ":tensorflow_lite_ops_inc_gen", |
| ":tensorflow_lite_structs_inc_gen", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_lib", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/core:framework", |
| "//tensorflow/lite/schema:schema_fbs", |
| "//third_party/eigen3", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:DerivedAttributeOpInterface", |
| "@llvm-project//mlir:Dialect", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:InferTypeOpInterface", |
| "@llvm-project//mlir:LoopLikeInterface", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:SideEffects", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:TransformUtils", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "variables_utils", |
| srcs = [ |
| "utils/variables_utils.cc", |
| ], |
| hdrs = [ |
| "utils/variables_utils.h", |
| ], |
| deps = [ |
| "//tensorflow/compiler/mlir/tensorflow", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:QuantOps", |
| ], |
| ) |
| |
| cc_library( |
| name = "cost_estimators", |
| hdrs = [ |
| "utils/arithmetic_count_util.h", |
| ], |
| deps = [ |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| cc_library( |
| name = "constant_utils", |
| srcs = [ |
| "utils/constant_utils.cc", |
| ], |
| hdrs = [ |
| "utils/constant_utils.h", |
| ], |
| copts = ["-std=c++14"], |
| deps = [ |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:mangling_util", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/stream_executor/lib", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| cc_library( |
| name = "lstm_utils", |
| srcs = [ |
| "utils/lstm_utils.cc", |
| ], |
| hdrs = [ |
| "utils/lstm_utils.h", |
| ], |
| copts = ["-std=c++14"], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:TensorDialect", |
| ], |
| ) |
| |
| cc_library( |
| name = "fake_quant_utils", |
| srcs = [ |
| "utils/fake_quant_utils.cc", |
| ], |
| hdrs = [ |
| "utils/fake_quant_utils.h", |
| ], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| cc_library( |
| name = "nms_utils", |
| srcs = [ |
| "utils/nms_utils.cc", |
| ], |
| hdrs = [ |
| "utils/nms_utils.h", |
| ], |
| copts = ["-std=c++14"], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_attributes", |
| "//tensorflow/core:framework", |
| "@flatbuffers", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| cc_library( |
| name = "tftext_utils", |
| srcs = [ |
| "utils/tftext_utils.cc", |
| ], |
| hdrs = [ |
| "utils/tftext_utils.h", |
| ], |
| copts = ["-std=c++14"], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_attributes", |
| "//tensorflow/core:framework", |
| "@flatbuffers", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "tftext_utils_test", |
| size = "small", |
| srcs = ["utils/tftext_utils_test.cc"], |
| deps = [ |
| ":tftext_utils", |
| "//tensorflow/core:framework", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| cc_library( |
| name = "perception_ops_utils", |
| srcs = [ |
| "utils/perception_ops_utils.cc", |
| ], |
| hdrs = [ |
| "utils/perception_ops_utils.h", |
| ], |
| copts = ["-std=c++14"], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_attributes", |
| "//tensorflow/lite/c:common", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| cc_library( |
| name = "stateful_ops_utils", |
| srcs = [ |
| "utils/stateful_ops_utils.cc", |
| ], |
| hdrs = [ |
| "utils/stateful_ops_utils.h", |
| ], |
| deps = [ |
| ":tensorflow_lite", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "lstm_utils_test", |
| size = "small", |
| srcs = ["utils/lstm_utils_test.cc"], |
| deps = [ |
| ":lstm_utils", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "perception_ops_utils_test", |
| size = "small", |
| srcs = ["utils/perception_ops_utils_test.cc"], |
| deps = [ |
| ":perception_ops_utils", |
| "//tensorflow/compiler/mlir/lite:tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_attributes", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:StandardOps", |
| ], |
| ) |
| |
| cc_library( |
| name = "tensorflow_lite_legalize_tf", |
| srcs = [ |
| "transforms/dilated_conv.cc", |
| "transforms/generated_legalize_tf.inc", |
| "transforms/generated_lower_static_tensor_list.inc", |
| "transforms/generated_prepare_tf.inc", |
| "transforms/get_arithmetic_count.cc", |
| "transforms/initialize_variables.cc", |
| "transforms/insert_call_once_op.cc", |
| "transforms/legalize_hashtables.cc", |
| "transforms/legalize_tf.cc", |
| "transforms/legalize_tf_while.cc", |
| "transforms/legalize_variables.cc", |
| "transforms/lower_static_tensor_list.cc", |
| "transforms/optimize_functional_ops.cc", |
| "transforms/prepare_composite_functions_tf.cc", |
| "transforms/prepare_tf.cc", |
| "transforms/raise_custom_ops.cc", |
| "transforms/remove_unused_func_args_tensors.cc", |
| "transforms/runtime_verify.cc", |
| "transforms/split_merged_operands.cc", |
| "transforms/trim_functions_tf.cc", |
| "transforms/while_loop_outline.cc", |
| ], |
| hdrs = [ |
| "ir/tfl_ops_interface.h.inc", |
| "transforms/dilated_conv.h", |
| "transforms/passes.h", |
| ], |
| deps = [ |
| ":constant_utils", |
| ":cost_estimators", |
| ":fake_quant_utils", |
| ":lstm_utils", |
| ":nms_utils", |
| ":perception_ops_utils", |
| ":stateful_ops_utils", |
| ":tensorflow_lite", |
| ":tftext_utils", |
| ":validators", |
| ":variables_utils", |
| "//tensorflow/compiler/mlir:op_or_arg_name_mapper", |
| "//tensorflow/compiler/mlir/hlo", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_lib", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:convert_tensor", |
| "//tensorflow/compiler/mlir/tensorflow:error_util", |
| "//tensorflow/compiler/mlir/tensorflow:lower_tf_lib", |
| "//tensorflow/compiler/mlir/tensorflow:mangling_util", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_attributes", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/compiler/mlir/tensorflow:tf_legalize_hlo", |
| "//tensorflow/compiler/mlir/tensorflow:unroll_batch_matmul_pass", |
| "//tensorflow/compiler/mlir/tensorflow:verification_utils", |
| "//tensorflow/compiler/mlir/xla:xla_legalize_tf", |
| "//tensorflow/compiler/mlir/xla:xla_legalize_tf_with_tf2xla", |
| "//tensorflow/compiler/xla:status", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:framework", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/kernels:tensor_list", |
| "@com_google_absl//absl/container:flat_hash_set", |
| "@com_google_absl//absl/container:inlined_vector", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/numeric:bits", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:Analysis", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:Transforms", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "tensorflow_lite_optimize", |
| srcs = [ |
| "transforms/generated_optimize.inc", |
| "transforms/optimize.cc", |
| ], |
| hdrs = [ |
| "transforms/passes.h", |
| ], |
| deps = [ |
| ":convert_type", |
| ":tensorflow_lite", |
| ":validators", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_lib", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:verification_utils", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:TransformUtils", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "tensorflow_lite_quantize", |
| srcs = [ |
| "transforms/default_quant_params.cc", |
| "transforms/generated_post_quantize.inc", |
| "transforms/generated_quantize.inc", |
| "transforms/modify_io_nodes.cc", |
| "transforms/optimize_op_order.cc", |
| "transforms/post_quantize.cc", |
| "transforms/prepare_quantize.cc", |
| "transforms/quantize.cc", |
| "utils/generated_op_quant_spec_getters.inc", |
| ], |
| hdrs = [ |
| "transforms/passes.h", |
| "transforms/prepare_quantize_helper.h", |
| ], |
| deps = [ |
| "convert_type", |
| ":tensorflow_lite", |
| ":validators", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_config", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_lib", |
| "//tensorflow/compiler/mlir/lite/quantization/lite:tfl_to_std", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/lite/schema:schema_fbs", |
| "//tensorflow/lite/tools/optimize:operator_property", |
| "@com_google_absl//absl/container:flat_hash_set", |
| "@com_google_absl//absl/memory", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:TransformUtils", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "tensorflow_lite_d2s", |
| srcs = [ |
| "transforms/dense_to_sparse.cc", |
| ], |
| hdrs = [ |
| "transforms/passes.h", |
| ], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/lite/tools/optimize/sparsity:format_converter", |
| "//third_party/eigen3", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:StandardOps", |
| ], |
| alwayslink = 1, |
| ) |
| |
| filegroup( |
| name = "generated_op_quant_spec_getters", |
| srcs = [ |
| "utils/generated_op_quant_spec_getters.inc", |
| ], |
| ) |
| |
| gentbl( |
| name = "op_quant_spec_getters_inc", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [("", "utils/generated_op_quant_spec_getters.inc")], |
| tblgen = "//tensorflow/compiler/mlir/lite/quantization:op_quant_spec_getters_gen", |
| td_file = "ir/tfl_ops.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| ], |
| ) |
| |
| tf_native_cc_binary( |
| name = "converter-gen", |
| srcs = [ |
| "converter_gen.cc", |
| ], |
| compatible_with = get_compatible_with_cloud(), |
| deps = [ |
| "@llvm-project//llvm:Support", |
| "@llvm-project//llvm:TableGen", |
| "@llvm-project//mlir:TableGen", |
| ], |
| ) |
| |
| gentbl( |
| name = "converter_inc", |
| compatible_with = get_compatible_with_cloud(), |
| tbl_outs = [ |
| ( |
| "--gen-operator-converters", |
| "operator_converters.inc", |
| ), |
| ( |
| "--gen-runtime-verifiers", |
| "runtime_verifiers.inc", |
| ), |
| ], |
| tblgen = ":converter-gen", |
| td_file = "ir/tfl_ops.td", |
| td_srcs = [ |
| ":tensorflow_lite_ops_td_files", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_td_files", |
| ], |
| test = 1, |
| ) |
| |
| cc_library( |
| name = "flatbuffer_tflite_operator_lib", |
| srcs = [ |
| "flatbuffer_operator.cc", |
| "operator_converters.inc", |
| ], |
| hdrs = [ |
| "flatbuffer_operator.h", |
| ], |
| deps = [ |
| ":convert_type", |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/lite/kernels/internal:kernel_utils", |
| "//tensorflow/lite/schema:schema_fbs", |
| "@com_google_absl//absl/container:flat_hash_map", |
| "@com_google_absl//absl/strings", |
| "@flatbuffers", |
| "@llvm-project//llvm:Analysis", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:TransformUtils", |
| ], |
| ) |
| |
| tf_native_cc_binary( |
| name = "flatbuffer_to_string", |
| srcs = ["flatbuffer_to_string.cc"], |
| deps = [ |
| "//tensorflow/lite/schema:schema_fbs_with_reflection", |
| "@flatbuffers", |
| ], |
| ) |
| |
| tf_native_cc_binary( |
| name = "json_to_flatbuffer", |
| srcs = ["json_to_flatbuffer.cc"], |
| deps = [ |
| "//tensorflow/lite/schema:schema_fbs", |
| "@flatbuffers", |
| ], |
| ) |
| |
| cc_library( |
| name = "emit_error_reporter", |
| srcs = [ |
| "emit_error_reporter.cc", |
| ], |
| hdrs = [ |
| "emit_error_reporter.h", |
| ], |
| deps = [ |
| "//tensorflow/lite/core/api", |
| "@llvm-project//mlir:IR", |
| ], |
| ) |
| |
| cc_library( |
| name = "flatbuffer_export", |
| srcs = [ |
| "flatbuffer_export.cc", |
| ], |
| hdrs = [ |
| "flatbuffer_export.h", |
| "flatbuffer_export_flags.h", |
| ], |
| deps = [ |
| ":convert_type", |
| ":flatbuffer_tflite_operator_lib", |
| ":stateful_ops_utils", |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir:op_or_arg_name_mapper", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:convert_tensor", |
| "//tensorflow/compiler/mlir/tensorflow:export_tf_dialect_op", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:framework", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/core/platform:logging", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/lite:schema_fbs_version", |
| "//tensorflow/lite:string_util", |
| "//tensorflow/lite/delegates/flex:allowlisted_flex_ops_lib", |
| "//tensorflow/lite/kernels/internal:kernel_utils", |
| "//tensorflow/lite/schema:schema_conversion_utils", |
| "//tensorflow/lite/schema:schema_fbs", |
| "//tensorflow/lite/tools/versioning", |
| "@com_google_absl//absl/base:core_headers", |
| "@com_google_absl//absl/container:flat_hash_map", |
| "@com_google_absl//absl/container:flat_hash_set", |
| "@com_google_absl//absl/strings", |
| "@flatbuffers", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:Translation", |
| ], |
| ) |
| |
| cc_library( |
| name = "flatbuffer_import", |
| srcs = [ |
| "flatbuffer_import.cc", |
| ], |
| hdrs = [ |
| "flatbuffer_import.h", |
| ], |
| deps = [ |
| ":convert_type", |
| ":flatbuffer_tflite_operator_lib", |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_lib", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:mangling_util", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/lite:framework", |
| "//tensorflow/lite/schema:schema_fbs", |
| "//tensorflow/lite/schema:schema_utils", |
| "@com_google_absl//absl/base", |
| "@com_google_absl//absl/container:flat_hash_map", |
| "@com_google_absl//absl/container:flat_hash_set", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:Translation", |
| ], |
| ) |
| |
| cc_library( |
| name = "convert_type", |
| srcs = [ |
| "utils/convert_type.cc", |
| ], |
| hdrs = [ |
| "utils/convert_type.h", |
| ], |
| deps = [ |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_types", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/lite/schema:schema_fbs", |
| "@llvm-project//mlir:IR", |
| ], |
| ) |
| |
| cc_library( |
| name = "flatbuffer_translate_lib", |
| hdrs = [ |
| "flatbuffer_export.h", |
| "flatbuffer_export_flags.h", |
| "flatbuffer_import.h", |
| "utils/convert_type.h", |
| ], |
| deps = [ |
| ":flatbuffer_export", |
| ":flatbuffer_import", |
| "//tensorflow/compiler/mlir:op_or_arg_name_mapper", |
| "//tensorflow/compiler/xla:statusor", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/lite/schema:schema_fbs", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//mlir:IR", |
| ], |
| ) |
| |
| cc_library( |
| name = "flatbuffer_translate_registeration", |
| srcs = [ |
| "flatbuffer_translate.cc", |
| ], |
| deps = [ |
| ":flatbuffer_translate_lib", |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:MlirTranslateMain", |
| "@llvm-project//mlir:QuantOps", |
| "@llvm-project//mlir:StandardOps", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:Translation", |
| ], |
| alwayslink = 1, |
| ) |
| |
| tf_cc_binary( |
| name = "flatbuffer_translate", |
| deps = [ |
| ":flatbuffer_translate_registeration", |
| # TODO(b/155809683): Link only necessary dialects. |
| "@llvm-project//mlir:AllPassesAndDialects", |
| ], |
| ) |
| |
| cc_library( |
| name = "tf_tfl_translate_cl_options", |
| srcs = [ |
| "tf_tfl_translate_cl.cc", |
| ], |
| hdrs = [ |
| "tf_tfl_translate_cl.h", |
| ], |
| deps = [ |
| "@llvm-project//llvm:Support", |
| ], |
| alwayslink = 1, |
| ) |
| |
| cc_library( |
| name = "common", |
| hdrs = [ |
| "common/tfl_pass_config.h", |
| ], |
| deps = [ |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_config", |
| "@llvm-project//llvm:Support", |
| ], |
| ) |
| |
| filegroup( |
| name = "tf_tfl_translate_main", |
| srcs = [ |
| "tf_tfl_translate.cc", |
| ], |
| ) |
| |
| tf_cc_binary( |
| name = "tf_tfl_translate", |
| srcs = [ |
| ":tf_tfl_translate_main", |
| ], |
| deps = [ |
| ":common", |
| ":flatbuffer_translate_lib", |
| ":flatbuffer_translate_registeration", |
| ":tensorflow_lite", |
| ":tf_tfl_passes", |
| ":tf_tfl_translate_cl_options", |
| ":tf_to_tfl_flatbuffer", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:Support", |
| # TODO(b/155809683): Link only necessary dialects. |
| "@llvm-project//mlir:AllPassesAndDialects", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:Support", |
| "//tensorflow/cc/saved_model:loader", |
| "//tensorflow/compiler/mlir:init_mlir", |
| "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", |
| "//tensorflow/compiler/mlir/tensorflow:translate_cl_options", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/lite:framework", |
| "//tensorflow/lite/schema:schema_fbs", |
| "//tensorflow/stream_executor/lib", |
| ], |
| ) |
| |
| tf_cc_binary( |
| name = "mlir-tflite-runner", |
| srcs = [ |
| "mlir_tflite_runner.cc", |
| ], |
| deps = [ |
| ":flatbuffer_translate_lib", |
| ":flatbuffer_translate_registeration", |
| ":tensorflow_lite", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/core:lib", |
| "//tensorflow/core/platform:logging", |
| "//tensorflow/lite:framework", |
| "//tensorflow/lite/delegates/flex:delegate", |
| "//tensorflow/lite/kernels:builtin_ops", |
| "@com_google_absl//absl/strings", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Parser", |
| "@llvm-project//mlir:StandardOps", |
| ], |
| ) |
| |
| cc_library( |
| name = "tf_tfl_passes", |
| srcs = ["tf_tfl_passes.cc"], |
| hdrs = [ |
| "tf_tfl_passes.h", |
| ], |
| deps = [ |
| ":common", |
| ":fake_quant_utils", |
| ":tensorflow_lite_d2s", |
| ":tensorflow_lite_legalize_tf", |
| ":tensorflow_lite_optimize", |
| ":tensorflow_lite_quantize", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_config", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_passes", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:decode_constant_pass", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops", |
| "//tensorflow/compiler/mlir/tensorflow:tensorflow_passes", |
| "//tensorflow/compiler/mlir/tensorflow:tf_graph_optimization_pass", |
| "//tensorflow/compiler/mlir/tensorflow:tf_saved_model_passes", |
| "//tensorflow/compiler/mlir/tensorflow:translate_lib", |
| "//tensorflow/core:core_cpu_base", |
| "//tensorflow/lite/toco:model_flags_proto_cc", |
| "//tensorflow/lite/toco:toco_flags_proto_cc", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:Transforms", |
| ], |
| ) |
| |
| cc_library( |
| name = "tf_to_tfl_flatbuffer", |
| srcs = ["tf_to_tfl_flatbuffer.cc"], |
| hdrs = [ |
| "tf_to_tfl_flatbuffer.h", |
| ], |
| deps = [ |
| ":common", |
| ":flatbuffer_translate_lib", |
| ":tensorflow_lite", |
| ":tensorflow_lite_legalize_tf", |
| ":tensorflow_lite_optimize", |
| ":tensorflow_lite_quantize", |
| "//tensorflow/compiler/mlir/lite/quantization:quantization_config", |
| "//tensorflow/compiler/mlir/tensorflow", |
| "//tensorflow/compiler/mlir/tensorflow:decode_constant_pass", |
| "//tensorflow/compiler/mlir/tensorflow:error_util", |
| "//tensorflow/compiler/mlir/tensorflow:mlir_roundtrip_flags", |
| "//tensorflow/compiler/mlir/tensorflow:tf_dialect_lib", |
| "//tensorflow/compiler/mlir/tensorflow:tf_dialect_passes", |
| "//tensorflow/compiler/mlir/tensorflow:translate_lib", |
| "//tensorflow/core:framework", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/lite/tools/optimize:quantize_weights", |
| "//tensorflow/stream_executor/lib", |
| "@com_google_absl//absl/types:span", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:AllPassesAndDialects", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Parser", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:Support", |
| "@llvm-project//mlir:Transforms", |
| ], |
| ) |
| |
| cc_library( |
| name = "empty_passes", |
| hdrs = ["transforms/passes.h"], |
| visibility = [ |
| "//configs/devtools/hawkeye/tflite:__subpackages__", |
| "//learning/brain/models/app_benchmarks:__subpackages__", |
| "//tensorflow/compiler/mlir/lite:friends", |
| "//tensorflow/lite/experimental/mlir:__subpackages__", |
| ], |
| deps = [ |
| "@llvm-project//llvm:Support", |
| ], |
| ) |
