diff options
| author |  cbockman <c.bockman@gmail.com> | 2018-08-03 17:05:39 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2018-08-03 17:05:39 -0700 |
| commit | eec7744f5a29411507ad7c7e5fd9ef19cf07c699 (patch) | |
| tree | 77f858d7adf96bfcfab0e51667df6899a3e89aca /tensorflow/contrib/crf | |
| parent | a2dc1d52ab068b531e85cabaf5043d920b56c0f4 (diff) | |
Change test cases to ensure survivable to different var types
Diffstat (limited to 'tensorflow/contrib/crf')
| -rw-r--r-- | tensorflow/contrib/crf/python/kernel_tests/crf_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/crf/python/kernel_tests/crf_test.py b/tensorflow/contrib/crf/python/kernel_tests/crf_test.py index f56a973f6f..8cfe142059 100644 --- a/tensorflow/contrib/crf/python/kernel_tests/crf_test.py +++ b/tensorflow/contrib/crf/python/kernel_tests/crf_test.py @@ -158,7 +158,7 @@ class CrfTest(test.TestCase): # Test both the length-1 and regular cases. sequence_lengths_list = [ np.array(3, dtype=np.int32), - np.array(1, dtype=np.int32) + np.array(1, dtype=np.int64) ] inputs_list = [ np.array([[4, 5, -3], [3, -1, 3], [-1, 2, 1], [0, 0, 0]], @@ -291,7 +291,7 @@ class CrfTest(test.TestCase): # Test both the length-1 and regular cases. sequence_lengths_list = [ np.array(3, dtype=np.int32), - np.array(1, dtype=np.int32) + np.array(1, dtype=np.int64) ] inputs_list = [ np.array([[4, 5, -3], [3, -1, 3], [-1, 2, 1], [0, 0, 0]], |