Skip to content

N
ngraph
Commit fb7e61da authored by Sergey Shalnov's avatar Sergey Shalnov Committed by Robert Kimball
Browse files
Options

IntelGPU backend: Relu operation fix with int32_t (#2596)

parent 8095c8ca
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 3 deletions
src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
View file @ fb7e61da
...@@ -1227,9 +1227,14 @@ shared_ptr<runtime::Executable> ...@@ -1227,9 +1227,14 @@ shared_ptr<runtime::Executable>
} }
case OP_TYPEID::Relu: case OP_TYPEID::Relu:
{ {
const string zero_const = const string output_type_name = get_opencl_type_name(get_output_type(op));
"convert_" + get_opencl_type_name(get_output_type(op)) + "(0)"; const string convert_to_type = "convert_" + output_type_name;
do_universal_unary(topology, op, "max(" + zero_const + ", input_var)", activation_relu); const string zero_const = convert_to_type + "(0)";
do_universal_unary(topology,
op,
"max(" + zero_const + ", " + convert_to_type + "(input_var))",
activation_relu);
break; break;
} }
case OP_TYPEID::Sigmoid: case OP_TYPEID::Sigmoid:
......
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