Skip to content

Commit

Permalink
Revert "Implement 3D and transposed 3D convolutions. (#1945)" + #1961
Browse files Browse the repository at this point in the history
This reverts commit d696d74.
  • Loading branch information
syl20bnr committed Jul 5, 2024
1 parent 51aea94 commit 6a09326
Show file tree
Hide file tree
Showing 73 changed files with 354 additions and 6,697 deletions.
9 changes: 0 additions & 9 deletions backend-comparison/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -72,19 +72,10 @@ name = "conv-transpose2d"
path = "benches/conv_transpose2d.rs"
harness = false

[[bench]]
name = "conv-transpose3d"
path = "benches/conv_transpose3d.rs"
harness = false

[[bench]]
name = "conv2d"
harness = false

[[bench]]
name = "conv3d"
harness = false

[[bench]]
name = "matmul"
harness = false
Expand Down
109 changes: 0 additions & 109 deletions backend-comparison/benches/conv3d.rs

This file was deleted.

111 changes: 0 additions & 111 deletions backend-comparison/benches/conv_transpose3d.rs

This file was deleted.

4 changes: 0 additions & 4 deletions backend-comparison/src/burnbenchapp/base.rs
Original file line number Diff line number Diff line change
Expand Up @@ -104,12 +104,8 @@ enum BenchmarkValues {
Autodiff,
#[strum(to_string = "conv-transpose2d")]
ConvTranspose2d,
#[strum(to_string = "conv-transpose3d")]
ConvTranspose3d,
#[strum(to_string = "conv2d")]
Conv2d,
#[strum(to_string = "conv3d")]
Conv3d,
}

pub fn execute() {
Expand Down
2 changes: 0 additions & 2 deletions burn-book/src/building-blocks/module.md
Original file line number Diff line number Diff line change
Expand Up @@ -183,10 +183,8 @@ Burn comes with built-in modules that you can use to build your own modules.
| ----------------- | -------------------- |
| `Conv1d` | `nn.Conv1d` |
| `Conv2d` | `nn.Conv2d` |
| `Conv3d` | `nn.Conv3d` |
| `ConvTranspose1d` | `nn.ConvTranspose1d` |
| `ConvTranspose2d` | `nn.ConvTranspose2d` |
| `ConvTranspose3d` | `nn.ConvTranspose3d` |

### Pooling

Expand Down
Loading

0 comments on commit 6a09326

Please sign in to comment.