diff options
| author |  Peter Hawkins <phawkins@google.com> | 2018-08-06 14:16:18 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-06 14:24:43 -0700 |
| commit | f3aa5d4717deac34b00979fdc4f13c3c9170e819 (patch) | |
| tree | acee6965459509da01a24a25241ffa5f19d6f057 /tensorflow/python/BUILD | |
| parent | 410094f7d392f3c0a384aa5caa310d9452ef6587 (diff) | |
[TF] Extend the Softmax kernels so they accept shapes with rank >= 1, rather than relying on the caller to reshape to rank 2.
Guard the Python library code that reshapes softmax inputs to rank 1 with a forward compatibility check; after the forward compatibility window expires, the Python code will no longer reshape to rank 1.
PiperOrigin-RevId: 207606326
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 614499793b..2e6fb11655 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3173,6 +3173,7 @@ cuda_py_test( ":partitioned_variables", ":variable_scope", ":variables", + "@absl_py//absl/testing:parameterized", "//third_party/py/numpy", ], tags = ["no_windows"], |