| # Copyright 2021 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| """Minimal set of aliases to redirect mlir-hlo dependencies to their location in TF.""" |
| |
| package( |
| default_visibility = ["//visibility:public"], |
| licenses = ["notice"], |
| ) |
| |
| alias( |
| name = "hlo", |
| actual = "@org_tensorflow//tensorflow/compiler/mlir/hlo", |
| ) |
| |
| [ |
| alias( |
| name = name, |
| actual = "@org_tensorflow//tensorflow/compiler/mlir/hlo:" + name, |
| ) |
| for name in [ |
| "chlo_legalize_to_hlo", |
| "legalize_control_flow", |
| "legalize_einsum_to_dot_general", |
| "legalize_gather_to_torch_index_select", |
| "legalize_to_linalg", |
| "legalize_to_standard", |
| "lhlo", |
| "map_lmhlo_to_scalar_op", |
| "map_mhlo_to_scalar_op", |
| "materialize_broadcasts", |
| "mhlo_control_flow_to_scf", |
| "mhlo_to_mhlo_lowering_patterns", |
| "unfuse_batch_norm", |
| ] |
| ] |
