Move library 'gradient_checker' out of kernel_tests/ into ops/.

Change: 125460867
This commit is contained in:
A. Unique TensorFlower 2016-06-21 08:43:07 -08:00 committed by TensorFlower Gardener
parent 1602ac6a91
commit 6290ac2c25
6 changed files with 17 additions and 12 deletions

View File

@ -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(

View File

@ -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",

View File

@ -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

View File

@ -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

View File

@ -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")