Commit 0cc2c500 authored by Cresson Remi's avatar Cresson Remi
Browse files

COMP: avoid warning

Showing with 3 additions and 1 deletion
+3 -1
......@@ -105,7 +105,9 @@ void GetTensorAttributes(const tensorflow::GraphDef & graph, std::vector<std::st
if (node.name().compare((*nameIt)) == 0)
{
found = true;
tensorflow::DataType ts_dt;
// Set default to DT_FLOAT
tensorflow::DataType ts_dt = tensorflow::DT_FLOAT;
// Default (input?) tensor type
auto test_is_output = node.attr().find("T");
......
Supports Markdown
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