Skip to content

GitLab

Commit b60895cd authored by turneram's avatar turneram
Browse files

Formatting

parent 5a2b89fc
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
src/simplify_algebra.cpp
View file @ b60895cd
...@@ -814,8 +814,8 @@ struct find_conv_dot_horiz_fusion ...@@ -814,8 +814,8 @@ struct find_conv_dot_horiz_fusion
m.move_instructions(arg, input); m.move_instructions(arg, input);
} }
// TODO: Check if axises match // TODO: Check if axises match
auto concat = auto concat = m.insert_instruction(
m.insert_instruction(input, make_op("concat", {{"axis", concat_axis}}), args); input, make_op("concat", {{"axis", concat_axis}}), args);
auto batch_size = input->get_shape().lens().front(); auto batch_size = input->get_shape().lens().front();
auto sequence_length = input->get_shape().lens().at(1); auto sequence_length = input->get_shape().lens().at(1);
auto hidden_size = input->get_shape().lens().at(2); auto hidden_size = input->get_shape().lens().at(2);
...@@ -840,10 +840,11 @@ struct find_conv_dot_horiz_fusion ...@@ -840,10 +840,11 @@ struct find_conv_dot_horiz_fusion
{{"dims", {batch_size, sequence_length, hidden_size * 3}}}), {{"dims", {batch_size, sequence_length, hidden_size * 3}}}),
fused); fused);
int64_t len = arg->get_shape().lens()[axis]; int64_t len = arg->get_shape().lens()[axis];
m.replace_instruction( m.replace_instruction(
arg, arg,
make_op("slice", make_op(
{{"axes", {axis}}, {"starts", {offset}}, {"ends", {offset + len}}}), "slice",
{{"axes", {axis}}, {"starts", {offset}}, {"ends", {offset + len}}}),
fused); fused);
offset += len; offset += len;
} }
......
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