32 lines
700 B
Python
32 lines
700 B
Python
load("//tensorflow:tensorflow.bzl", "cuda_py_test")
|
|
|
|
package(
|
|
default_visibility = ["//visibility:private"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
py_library(
|
|
name = "dlpack",
|
|
srcs = ["dlpack.py"],
|
|
srcs_version = "PY3",
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
deps = [
|
|
"//tensorflow/python:pywrap_tensorflow",
|
|
],
|
|
)
|
|
|
|
cuda_py_test(
|
|
name = "dlpack_test",
|
|
srcs = ["dlpack_test.py"],
|
|
srcs_version = "PY2AND3",
|
|
tags = [
|
|
"no_tfrt", # TODO(b/177356372)
|
|
],
|
|
deps = [
|
|
":dlpack",
|
|
"//tensorflow/python/eager:test",
|
|
"@absl_py//absl/testing:absltest",
|
|
"@absl_py//absl/testing:parameterized",
|
|
],
|
|
)
|