提交 4c8b4536 编辑于 作者: Michael Reneer's avatar Michael Reneer 提交者: tensorflow-copybara
浏览文件

Add tracing API to the Python package.

PiperOrigin-RevId: 342128507
上级 870441c8
......@@ -26,6 +26,7 @@ py_library(
deps = [
":version",
"//tensorflow_federated/python/aggregators",
"//tensorflow_federated/python/common_libs:tracing",
"//tensorflow_federated/python/core/api:computation_base",
"//tensorflow_federated/python/core/api:computation_types",
"//tensorflow_federated/python/core/api:computations",
......
......@@ -20,6 +20,7 @@ from tensorflow_federated.version import __version__ # pylint: disable=g-bad-im
from tensorflow_federated.python import aggregators
from tensorflow_federated.python import learning
from tensorflow_federated.python import simulation
from tensorflow_federated.python.common_libs import tracing as profiler
from tensorflow_federated.python.core import backends
from tensorflow_federated.python.core import framework
from tensorflow_federated.python.core import templates
......
......@@ -126,8 +126,10 @@ py_library(
name = "tracing",
srcs = ["tracing.py"],
srcs_version = "PY3",
# TODO(b/163856734): Fix the visibility of the `common_libs`.
visibility = ["//tensorflow_federated:internal"],
visibility = [
"//tensorflow_federated:__pkg__",
"//tensorflow_federated:internal",
],
deps = [":py_typecheck"],
)
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册