• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Description:
2#   Contains the Keras wrapper API (internal TensorFlow version).
3
4load("//tensorflow:tensorflow.bzl", "tf_py_test")
5
6package(
7    default_visibility = ["//tensorflow/python/keras:__subpackages__"],
8    licenses = ["notice"],  # Apache 2.0
9)
10
11filegroup(
12    name = "all_py_srcs",
13    srcs = glob(["*.py"]),
14    visibility = ["//tensorflow/python/keras/google/private_tf_api_test:__pkg__"],
15)
16
17py_library(
18    name = "wrappers",
19    srcs = [
20        "__init__.py",
21        "scikit_learn.py",
22    ],
23    srcs_version = "PY3",
24    deps = [
25        "//tensorflow/python:util",
26        "//tensorflow/python/keras:engine",
27        "//tensorflow/python/keras:losses",
28        "//tensorflow/python/keras/utils:generic_utils",
29        "//third_party/py/numpy",
30    ],
31)
32
33tf_py_test(
34    name = "scikit_learn_test",
35    size = "small",
36    srcs = ["scikit_learn_test.py"],
37    python_version = "PY3",
38    tags = ["notsan"],
39    deps = [
40        ":wrappers",
41        "//tensorflow/python:client_testlib",
42        "//tensorflow/python/keras:testing_utils",
43        "//tensorflow/python/keras/utils:np_utils",
44        "//third_party/py/numpy",
45    ],
46)
47