Commit 98510d96 authored by remi cresson's avatar remi cresson
Browse files

ADD: Tensorflow::DT_BOOL type in data bridge

parent e9ed30fb
...@@ -19,7 +19,11 @@ namespace tf { ...@@ -19,7 +19,11 @@ namespace tf {
template<class Type> template<class Type>
tensorflow::DataType GetTensorflowDataType() 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; return tensorflow::DT_INT8;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment