diff options
author | 2018-03-14 20:07:38 -0700 | |
---|---|---|
committer | 2018-03-14 20:11:38 -0700 | |
commit | 8b305c509353815e2086200b47d0af898fa9df35 (patch) | |
tree | 76cc456451f6269deea8ca46925693cfa4e72794 | |
parent | 6b2e5a4a1a11f3bed1260bf63cf78af3e7109a9d (diff) |
Add int64 to randomized tests.
PiperOrigin-RevId: 189129641
-rw-r--r-- | tensorflow/compiler/tests/randomized_tests.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/tests/randomized_tests.cc b/tensorflow/compiler/tests/randomized_tests.cc index e72dd4eea9..e53efc3091 100644 --- a/tensorflow/compiler/tests/randomized_tests.cc +++ b/tensorflow/compiler/tests/randomized_tests.cc @@ -83,8 +83,8 @@ string LocalDeviceToFullDeviceName(const string& device) { return strings::StrCat("/job:localhost/replica:0/task:0/device:", device); } -constexpr std::array<DataType, 4> kAllXlaTypes = { - {DT_INT32, DT_FLOAT, DT_BOOL, DT_COMPLEX64}}; +constexpr std::array<DataType, 5> kAllXlaTypes = { + {DT_INT32, DT_FLOAT, DT_BOOL, DT_COMPLEX64, DT_INT64}}; // An OpTestBuilder is a graph builder class that takes as input an operator to // test, its inputs and attributes, and builds a graph that executes the |