1load("//tensorflow:tensorflow.bzl", "py_test") 2 3package( 4 default_visibility = ["//visibility:public"], 5 licenses = ["notice"], 6) 7 8py_library( 9 name = "mlir", 10 srcs = ["mlir.py"], 11 srcs_version = "PY3", 12 deps = [ 13 "//tensorflow/python:pywrap_mlir", 14 "//tensorflow/python/util:tf_export", 15 ], 16) 17 18py_test( 19 name = "mlir_test", 20 srcs = ["mlir_test.py"], 21 python_version = "PY3", 22 deps = [ 23 ":mlir", 24 "//tensorflow/python:client_testlib", 25 "//tensorflow/python:dtypes", 26 "//tensorflow/python:errors", 27 "//tensorflow/python:logging_ops", 28 "//tensorflow/python:tensor_spec", 29 "//tensorflow/python/eager:def_function", 30 ], 31) 32