79678a310f
Specifically, the constant is quantized according to the quantization information and dequantize ops is implemented by the xla dequantize ops. PiperOrigin-RevId: 293222337 Change-Id: If73ce832b35ed9e687c3ae28b506b7e12d3ea115
41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
package(
|
|
default_visibility = [
|
|
":friends",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
package_group(
|
|
name = "friends",
|
|
includes = ["//third_party/mlir:subpackages"],
|
|
packages = [
|
|
"//tensorflow/compiler/mlir/...",
|
|
"//tensorflow/compiler/mlir/lite/...",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "hlo_xla_quantization_passes",
|
|
srcs = [
|
|
"materialize.cc",
|
|
"op_quant_spec.inc",
|
|
"propagate.cc",
|
|
],
|
|
hdrs = [
|
|
"passes.h",
|
|
],
|
|
deps = [
|
|
"//tensorflow/compiler/mlir/lite/quantization:quantization_config",
|
|
"//tensorflow/compiler/mlir/lite/quantization:quantization_lib",
|
|
"//tensorflow/compiler/mlir/xla:hlo",
|
|
"//tensorflow/compiler/xla/client/lib:quantize",
|
|
"@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",
|
|
],
|
|
alwayslink = 1,
|
|
)
|