Skip to content

N
ngraph
Commit 8df14206 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by Robert Kimball
Browse files
Options

argmin/max fix (#1922)

parent fb3f9e95
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
src/ngraph/runtime/interpreter/int_backend.cpp
View file @ 8df14206
......@@ -178,7 +178,9 @@ bool runtime::interpreter::INTBackend::call(shared_ptr<Function> function,
{
case OP_TYPEID::Convert:
case OP_TYPEID::Quantize:
case OP_TYPEID::Dequantize: type = op->get_input_element_type(0); break;
case OP_TYPEID::Dequantize:
case OP_TYPEID::ArgMin:
case OP_TYPEID::ArgMax: type = op->get_input_element_type(0); break;
case OP_TYPEID::Equal:
case OP_TYPEID::Greater:
case OP_TYPEID::GreaterEq:
......
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