Skip to content

GitLab

Commit ba0b3794 authored by Alan Turner's avatar Alan Turner
Browse files

Formatting

parent 3058ef72
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
src/onnx/parse_where.cpp
View file @ ba0b3794
...@@ -56,10 +56,11 @@ struct parse_where : op_parser<parse_where> ...@@ -56,10 +56,11 @@ struct parse_where : op_parser<parse_where>
auto lens = auto lens =
compute_broadcasted_lens(args[0]->get_shape().lens(), args[1]->get_shape().lens()); compute_broadcasted_lens(args[0]->get_shape().lens(), args[1]->get_shape().lens());
lens = compute_broadcasted_lens(lens, args[2]->get_shape().lens()); lens = compute_broadcasted_lens(lens, args[2]->get_shape().lens());
// Convert condition tensor to int32 to work around CK not supporting bool type // Convert condition tensor to int32 to work around CK not supporting bool type
args[0] = info.add_instruction(make_op("convert", {{"target_type", shape::int32_type}}), args[0]); args[0] = info.add_instruction(make_op("convert", {{"target_type", shape::int32_type}}),
args[0]);
if(args[0]->get_shape().lens() != lens) if(args[0]->get_shape().lens() != lens)
{ {
args[0] = args[0] =
......
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