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

Handle DT_UINT32 & DT_UINT64 in transpose

Browse Source 
PiperOrigin-RevId: 317485472
Change-Id: I5c62d7977f44f57f544ebe27e82f141fe07ce73e
This commit is contained in:
Gaurav Jain 2020-06-20 14:12:04 -07:00 committed by TensorFlower Gardener
parent 7e0b098690
commit 059c05e4da
1 changed files with 2 additions and 0 deletions

2
tensorflow/core/kernels/transpose_functor.h
View File

@ -191,11 +191,13 @@ Status DoTransposeImpl(const Device& d, const Tensor& in,
case DT_FLOAT:
case DT_INT32:
case DT_QINT32:
case DT_UINT32:
Transpose<Device, uint32>::run(d, in, perm, out);
break;
case DT_DOUBLE:
case DT_INT64:
case DT_UINT64:
Transpose<Device, uint64>::run(d, in, perm, out);
break;