ENH: add uint8, uint16, int16, uint32, int64

c7b82461 · Cresson Remi · 56a98698 · c7b82461
Commit c7b82461 authored 4 years ago by Cresson Remi
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 2 deletions
+2 -2
--- a/include/otbTensorflowCopyUtils.cxx
+++ b/include/otbTensorflowCopyUtils.cxx
@@ -101,9 +101,9 @@ void RecopyImageRegionToTensorWithCast(const typename TImage::Pointer inputPtr,
  else if (dt == tensorflow::DT_DOUBLE)
    RecopyImageRegionToTensor<TImage, double>(inputPtr, region, tensor, elemIdx);
  else if (dt == tensorflow::DT_UINT64)
-    RecopyImageRegionToTensor<TImage, unsigned long int>(inputPtr, region, tensor, elemIdx);
+    RecopyImageRegionToTensor<TImage, unsigned long long int>(inputPtr, region, tensor, elemIdx);
  else if (dt == tensorflow::DT_INT64)
-    RecopyImageRegionToTensor<TImage, long int>(inputPtr, region, tensor, elemIdx);
+    RecopyImageRegionToTensor<TImage, long long int>(inputPtr, region, tensor, elemIdx);
  else if (dt == tensorflow::DT_UINT32)
    RecopyImageRegionToTensor<TImage, unsigned int>(inputPtr, region, tensor, elemIdx);
  else if (dt == tensorflow::DT_INT32)