From 059c05e4da07225faa651c46df9b9be6f83e6914 Mon Sep 17 00:00:00 2001
From: Gaurav Jain <gjn@google.com>
Date: Sat, 20 Jun 2020 14:12:04 -0700
Subject: [PATCH] Handle DT_UINT32 & DT_UINT64 in transpose
PiperOrigin-RevId: 317485472
Change-Id: I5c62d7977f44f57f544ebe27e82f141fe07ce73e
---
tensorflow/core/kernels/transpose_functor.h | 2 ++
1 file changed, 2 insertions(+)
diff --git a/tensorflow/core/kernels/transpose_functor.h b/tensorflow/core/kernels/transpose_functor.h
index a89fc40d772..0c22b11b7c6 100644
--- a/tensorflow/core/kernels/transpose_functor.h
+++ b/tensorflow/core/kernels/transpose_functor.h
@@ -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;