1load("//tensorflow:tensorflow.bzl", "tf_native_cc_binary") 2load( 3 "//tensorflow/core/platform:build_config.bzl", 4 "tf_proto_library", 5) 6 7package( 8 default_visibility = [ 9 ":friends", 10 ], 11 licenses = ["notice"], # Apache 2.0 12) 13 14package_group( 15 name = "friends", 16 includes = ["//third_party/mlir:subpackages"], 17 packages = ["//tensorflow/compiler/mlir/..."], 18) 19 20exports_files([ 21 "quantization_traits.h", 22 "quantization_config.h", 23]) 24 25filegroup( 26 name = "quantization_td_files", 27 srcs = [ 28 "quantization.td", 29 "@llvm-project//mlir:OpBaseTdFiles", 30 "@llvm-project//mlir:QuantizationOpsTdFiles", 31 ], 32) 33 34tf_proto_library( 35 name = "quantization_info_proto", 36 srcs = [ 37 "quantization_info.proto", 38 ], 39 cc_api_version = 2, 40) 41 42cc_library( 43 name = "quantization_passes", 44 srcs = [ 45 "import_quant_stats_pass.cc", 46 ], 47 hdrs = [ 48 "quantization_passes.h", 49 ], 50 deps = [ 51 ":quantization_info_proto_cc", 52 "//tensorflow/compiler/mlir/tensorflow:import_utils", 53 "//tensorflow/core:lib_proto_parsing", 54 "@com_google_absl//absl/memory", 55 "@com_google_absl//absl/strings", 56 "@llvm-project//llvm:support", 57 "@llvm-project//mlir:Analysis", 58 "@llvm-project//mlir:IR", 59 "@llvm-project//mlir:Pass", 60 "@llvm-project//mlir:QuantOps", 61 "@llvm-project//mlir:StandardOps", 62 "@llvm-project//mlir:Support", 63 ], 64 alwayslink = 1, 65) 66 67cc_library( 68 name = "quantization_lib", 69 srcs = [ 70 "quantization_driver.cc", 71 "quantization_utils.cc", 72 ], 73 hdrs = [ 74 "quantization_traits.h", 75 "quantization_utils.h", 76 ], 77 deps = [ 78 "//tensorflow/core:lib_proto_parsing", 79 "@com_google_absl//absl/memory", 80 "@llvm-project//llvm:support", 81 "@llvm-project//mlir:IR", 82 "@llvm-project//mlir:QuantOps", 83 "@llvm-project//mlir:StandardOps", 84 "@llvm-project//mlir:Support", 85 ], 86) 87 88cc_library( 89 name = "quantization_config", 90 srcs = [ 91 "quantization_config.cc", 92 ], 93 hdrs = [ 94 "quantization_config.h", 95 ], 96 deps = [ 97 "//tensorflow/core:protos_all_cc", 98 "@com_google_absl//absl/strings", 99 "@llvm-project//llvm:support", 100 ], 101) 102 103tf_native_cc_binary( 104 name = "op_quant_spec_getters_gen", 105 srcs = [ 106 "tools/op_quant_spec_getters_gen.cc", 107 ], 108 deps = [ 109 "@llvm-project//llvm:support", 110 "@llvm-project//llvm:tablegen", 111 "@llvm-project//mlir:TableGen", 112 ], 113) 114