| load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") |
| load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable") |
| load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library") |
| |
| package( |
| default_visibility = [ |
| "//tensorflow:internal", |
| "@tf_runtime//:friends", |
| ], |
| licenses = ["notice"], |
| ) |
| |
| gentbl_cc_library( |
| name = "rt_transforms_passes_inc_gen", |
| compatible_with = get_compatible_with_portable(), |
| tbl_outs = [ |
| ( |
| [ |
| "-gen-pass-decls", |
| "-name=RuntimeTransforms", |
| ], |
| "rt_gen_passes.h.inc", |
| ), |
| ], |
| tblgen = "@llvm-project//mlir:mlir-tblgen", |
| td_file = "rt_passes.td", |
| deps = ["@llvm-project//mlir:PassBaseTdFiles"], |
| ) |
| |
| cc_library( |
| name = "rt_transforms", |
| srcs = ["rt_convert_to_entrypoint.cc"], |
| hdrs = ["rt_passes.h"], |
| compatible_with = get_compatible_with_portable(), |
| deps = [ |
| ":rt_transforms_passes_inc_gen", |
| "//tensorflow/compiler/xla/mlir/ir/runtime:rt_ops", |
| "@llvm-project//llvm:Support", |
| "@llvm-project//mlir:ControlFlowDialect", |
| "@llvm-project//mlir:FuncDialect", |
| "@llvm-project//mlir:IR", |
| "@llvm-project//mlir:Pass", |
| "@llvm-project//mlir:Support", |
| ], |
| ) |