ADD: Tensorflow::DT_BOOL type in data bridge

98510d96 · remi cresson · e9ed30fb · 98510d96
Commit 98510d96 authored Jul 31, 2018 by remi cresson
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

include/otbTensorflowDataTypeBridge.cxx include/otbTensorflowDataTypeBridge.cxx +5 -1

No files found.
--- a/include/otbTensorflowDataTypeBridge.cxx
+++ b/include/otbTensorflowDataTypeBridge.cxx
@@ -19,7 +19,11 @@ namespace tf {
 template<class Type>
 tensorflow::DataType GetTensorflowDataType()
 {
-  if (typeid(Type) == typeid(char))
+  if (typeid(Type) == typeid(bool))
+  {
+    return tensorflow::DT_BOOL;
+  }
+  else if (typeid(Type) == typeid(char))
  {
    return tensorflow::DT_INT8;
  }