experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change int64 to int64_t to support 64-bit compilation

Browse Source 
PiperOrigin-RevId: 180957437
This commit is contained in:
Austin Anderson 2018-01-05 12:02:27 -08:00 committed by TensorFlower Gardener
parent ef7854f8a0
commit 388c9b7331
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/kernels/quantization_utils_test.cc
View File

@ -743,7 +743,8 @@ template <int POW>
void TestDivide64x2Pow(int64 val, int64 ref) {
const int64x2_t val_64x2 = vmovq_n_s64(val);
const int64x2_t ret = Divide64x2Pow<POW>(val_64x2);
int64 rets[2];
// TODO(b/70947959) Change back to int64 when possible
int64_t rets[2];
vst1q_s64(rets, ret);
EXPECT_EQ(rets[0], ref);
EXPECT_EQ(rets[1], ref);
@ -754,7 +755,8 @@ template <int POW>
void TestDivide64x2PowRound(int64 val, int64 ref) {
const int64x2_t val_64x2 = vmovq_n_s64(val);
const int64x2_t shifted = Divide64x2PowRound<POW>(val_64x2);
int64 rets[2];
// TODO(b/70947959) Change back to int64 when possible
int64_t rets[2];
vst1q_s64(rets, shifted);
EXPECT_EQ(rets[0], ref) << "in = " << val << ", " << POW
<< ", act = " << rets[0] << ", ref = " << ref;