diff options
author | 2018-06-20 13:19:50 -0700 | |
---|---|---|
committer | 2018-07-06 13:19:38 -0700 | |
commit | 1107fb018307dfdc35fbe1c2d2f2a378c45aeb18 (patch) | |
tree | f5a808c5216cef56d54f57484b14954bfcc5945c /tensorflow/core/kernels/cast_op_test.cc | |
parent | b2fe2a874bade4782aaca5c44bf29e7ff6c39200 (diff) |
Cast: support casting to and from quantized types
Diffstat (limited to 'tensorflow/core/kernels/cast_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op_test.cc | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cast_op_test.cc b/tensorflow/core/kernels/cast_op_test.cc index 7da9d28a3d..b74dc25837 100644 --- a/tensorflow/core/kernels/cast_op_test.cc +++ b/tensorflow/core/kernels/cast_op_test.cc @@ -76,7 +76,12 @@ class CastOpTest : public OpsTestBase { TEST_CAST(in, half); \ TEST_CAST(in, float); \ TEST_CAST(in, double); \ - TEST_CAST(in, bfloat16); + TEST_CAST(in, bfloat16); \ + TEST_CAST(in, quint8); \ + TEST_CAST(in, qint8); \ + TEST_CAST(in, qint32); \ + TEST_CAST(in, qint16); \ + TEST_CAST(in, quint16); TEST_ALL_CASTS_FROM(uint8) TEST_ALL_CASTS_FROM(uint16) @@ -87,6 +92,11 @@ TEST_ALL_CASTS_FROM(half) TEST_ALL_CASTS_FROM(float) TEST_ALL_CASTS_FROM(double) TEST_ALL_CASTS_FROM(bfloat16) +TEST_ALL_CASTS_FROM(quint8) +TEST_ALL_CASTS_FROM(qint8) +TEST_ALL_CASTS_FROM(qint32) +TEST_ALL_CASTS_FROM(qint16) +TEST_ALL_CASTS_FROM(quint16) #undef TEST_ALL_CASTS_FROM #undef TEST_CAST |