Docs.rs
diffusion_rs_common-0.1.0
diffusion_rs_common 0.1.0
Docs.rs crate page
MIT
Links
Homepage
Repository
crates.io
Source
Owners
EricLBuehler
Dependencies
accelerate-src ^0.3.2
normal
optional
anyhow ^1.0.94
normal
byteorder ^1.5.0
normal
cudarc ^0.12.1
normal
optional
dirs ^5.0.1
normal
float8 ^0.1.0
normal
gemm ^0.17.0
normal
half ^2.4.1
normal
hf-hub ^0.3.2
normal
indicatif ^0.17.9
normal
intel-mkl-src ^0.8.1
normal
optional
libc ^0.2.147
normal
optional
memmap2 ^0.9.5
normal
metal ^0.27.0
normal
optional
num-traits ^0.2.15
normal
num_cpus ^1.15.0
normal
rand ^0.8.5
normal
rand_distr ^0.4.3
normal
rayon ^1.1.0
normal
safetensors ^0.4.1
normal
serde ^1.0.215
normal
serde_json ^1.0.133
normal
thiserror ^2.0.4
normal
tokenizers ^0.21.0
normal
tqdm ^0.7.0
normal
ug ^0.0.2
normal
ug-cuda ^0.0.2
normal
optional
ug-metal ^0.0.2
normal
optional
yoke ^0.7.2
normal
zip ^2.2.1
normal
bindgen_cuda ^0.1.5
build
optional
Versions
17.76%
of the crate is documented
Platform
x86_64-unknown-linux-gnu
Feature flags
docs.rs
About docs.rs
Badges
Builds
Metadata
Shorthand URLs
Download
Rustdoc JSON
Build queue
Privacy policy
Rust
Rust website
The Book
Standard Library API Reference
Rust by Example
The Cargo Guide
Clippy Documentation
Module conv
diffusion_
rs_
common
0.1.0
Module conv
Module Items
Structs
Functions
In diffusion_
rs_
common::
nn
diffusion_rs_common
::
nn
Module
conv
Copy item path
Source
Expand description
Convolution Layers.
Structs
§
Conv1d
Conv1d
Config
Conv2d
Conv2d
Config
Conv
Transpose1d
Conv
Transpose1d
Config
Conv
Transpose2d
Conv
Transpose2d
Config
Functions
§
conv1d
conv1d_
no_
bias
conv2d
conv2d_
no_
bias
conv_
transpose1d
conv_
transpose1d_
no_
bias
conv_
transpose2d
conv_
transpose2d_
no_
bias