Trait dfdx::nn::BuildOnDevice
source · pub trait BuildOnDevice<D: Device<E>, E: Dtype> {
type Built: BuildModule<D, E>;
// Provided methods
fn build_on_device(device: &D) -> Self::Built { ... }
fn try_build_on_device(device: &D) -> Result<Self::Built, D::Err> { ... }
}
Expand description
Something that can be built on a different device than it is on.
Related to BuildModule