Move library 'gradient_checker' out of kernel_tests/ into ops/.
Change: 125460867
This commit is contained in:
parent
1602ac6a91
commit
6290ac2c25
@ -29,12 +29,12 @@ py_library(
|
||||
":client_testlib",
|
||||
":framework",
|
||||
":framework_test_lib",
|
||||
":gradient_checker",
|
||||
":platform",
|
||||
":platform_test",
|
||||
":summary",
|
||||
":training",
|
||||
"//tensorflow/contrib:contrib_py",
|
||||
"//tensorflow/python/kernel_tests:gradient_checker",
|
||||
],
|
||||
)
|
||||
|
||||
@ -856,6 +856,18 @@ py_library(
|
||||
],
|
||||
)
|
||||
|
||||
py_library(
|
||||
name = "gradient_checker",
|
||||
srcs = [
|
||||
"ops/gradient_checker.py",
|
||||
],
|
||||
srcs_version = "PY2AND3",
|
||||
deps = [
|
||||
":framework",
|
||||
":ops",
|
||||
],
|
||||
)
|
||||
|
||||
py_library(
|
||||
name = "training",
|
||||
srcs = glob(
|
||||
|
@ -13,12 +13,6 @@ licenses(["notice"]) # Apache 2.0
|
||||
load("//tensorflow:tensorflow.bzl", "py_tests")
|
||||
load("//tensorflow:tensorflow.bzl", "cuda_py_tests")
|
||||
|
||||
py_library(
|
||||
name = "gradient_checker",
|
||||
srcs = ["gradient_checker.py"],
|
||||
srcs_version = "PY2AND3",
|
||||
)
|
||||
|
||||
py_tests(
|
||||
name = "cpu_only_kernel_tests",
|
||||
size = "small",
|
||||
@ -109,7 +103,6 @@ cuda_py_tests(
|
||||
"functional_ops_test.py",
|
||||
"gather_nd_op_test.py",
|
||||
"gather_op_test.py",
|
||||
"gradient_checker_test.py",
|
||||
"gradient_correctness_test.py",
|
||||
"init_ops_test.py",
|
||||
"linalg_ops_test.py",
|
||||
|
@ -13,7 +13,7 @@
|
||||
# limitations under the License.
|
||||
# ==============================================================================
|
||||
|
||||
"""Tests for tensorflow.kernels.gradient_checker."""
|
||||
"""Tests for tf.test.compute_gradient and tf.compute_gradient_error."""
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
@ -67,8 +67,8 @@ from tensorflow.python.util.all_util import make_all
|
||||
from tensorflow.python.framework.test_util import TensorFlowTestCase as TestCase
|
||||
from tensorflow.python.framework.test_util import assert_equal_graph_def
|
||||
|
||||
from tensorflow.python.kernel_tests.gradient_checker import compute_gradient_error
|
||||
from tensorflow.python.kernel_tests.gradient_checker import compute_gradient
|
||||
from tensorflow.python.ops.gradient_checker import compute_gradient_error
|
||||
from tensorflow.python.ops.gradient_checker import compute_gradient
|
||||
# pylint: enable=unused-import
|
||||
|
||||
|
||||
|
@ -689,7 +689,7 @@ def tf_py_test(name, srcs, size="medium", data=[], main=None, args=[],
|
||||
data=data,
|
||||
deps=[
|
||||
"//tensorflow/python:extra_py_tests_deps",
|
||||
"//tensorflow/python/kernel_tests:gradient_checker",
|
||||
"//tensorflow/python:gradient_checker",
|
||||
] + additional_deps,
|
||||
srcs_version="PY2AND3")
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user