1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
cc_library(
name = "tosa_checker_lib",
srcs = ["tosa_checker.cc"],
hdrs = ["tosa_checker.h"],
deps = [
"@llvm-project//mlir:MlirTranslateMain",
"@org_tensorflow//tensorflow/compiler/mlir/lite:flatbuffer_translate_lib",
"@org_tensorflow//tensorflow/compiler/mlir/tosa:tfl_passes",
"@pybind11",
],
)
pybind_extension(
name = "_tosa_checker_wrapper",
srcs = [
"tosa_checker_pybind11.cc",
],
deps = [
":tosa_checker_lib",
],
)
py_library(
name = "tosa_checker",
srcs = [
"__init__.py",
],
srcs_version = "PY3",
visibility = ["//visibility:public"],
data = ["//tosa_checker:_tosa_checker_wrapper.so"],
)
|
