Fix output element type for selection functions (#1629)

* Fix output element type for selection functions * style

Fix output element type for selection functions (#1629)
* Fix output element type for selection functions * style
8970915b · Jaikrishnan Menon · Robert Kimball · c23a972d · 8970915b
Commit 8970915b authored Sep 20, 2018 by Jaikrishnan Menon Committed by Robert Kimball Sep 20, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

select_and_scatter.cpp src/ngraph/runtime/cpu/builder/select_and_scatter.cpp +2 -1

No files found.
--- a/src/ngraph/runtime/cpu/builder/select_and_scatter.cpp
+++ b/src/ngraph/runtime/cpu/builder/select_and_scatter.cpp
@@ -82,7 +82,8 @@ namespace ngraph
                    char output;
                    inputs.emplace_back(backend->create_tensor(element::f32, Shape{}, &x));
                    inputs.emplace_back(backend->create_tensor(element::f32, Shape{}, &y));
-                    outputs.emplace_back(backend->create_tensor(element::f32, Shape{}, &output));
+                    outputs.emplace_back(
+                        backend->create_tensor(element::boolean, Shape{}, &output));
                    select_external_function->make_call_frame()->call(outputs, inputs);
                    return output;
                };