"src/vscode:/vscode.git/clone" did not exist on "a671ef89ea4cc131f30d7b0fa5d4651c187e6f0b"
Commit 1555f298 authored by Paul's avatar Paul
Browse files

Format

parent 9288a1fe
...@@ -700,7 +700,7 @@ struct find_transpose_slice ...@@ -700,7 +700,7 @@ struct find_transpose_slice
[](auto x) { return x == 1; })) [](auto x) { return x == 1; }))
return; return;
// Compute axis before transpose to use for unsqueeeze // Compute axis before transpose to use for unsqueeeze
auto perm = ins->get_operator().to_value()["permutation"].to_vector<int64_t>(); auto perm = ins->get_operator().to_value()["permutation"].to_vector<int64_t>();
auto preaxis = std::find(perm.begin(), perm.end(), axis) - perm.begin(); auto preaxis = std::find(perm.begin(), perm.end(), axis) - perm.begin();
// Make unsqeeze // Make unsqeeze
auto unsqueeze = m.insert_instruction( auto unsqueeze = m.insert_instruction(
...@@ -711,7 +711,7 @@ struct find_transpose_slice ...@@ -711,7 +711,7 @@ struct find_transpose_slice
return i + 1; return i + 1;
return i; return i;
}); });
perm.insert(perm.begin(), preaxis+1); perm.insert(perm.begin(), preaxis + 1);
auto transpose = auto transpose =
m.insert_instruction(ins, make_op("transpose", {{"permutation", perm}}), unsqueeze); m.insert_instruction(ins, make_op("transpose", {{"permutation", perm}}), unsqueeze);
// Slice and sqeeze // Slice and sqeeze
......
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