Index

_

_compute_max_input_range() (distdl.nn.mixins.ConvMixin method) (distdl.nn.mixins.PoolingMixin method) _compute_min_input_range() (distdl.nn.mixins.ConvMixin method) (distdl.nn.mixins.PoolingMixin method)

_distdl_forward_pre_hook() (distdl.nn.module.Module static method) _distdl_input_changed() (distdl.nn.module.Module method) _distdl_module_requires_reset() (distdl.nn.module.Module method) _distdl_module_setup() (distdl.nn.module.Module method) _distdl_module_teardown() (distdl.nn.module.Module method)

A

allgather_data() (distdl.backends.mpi.partition.MPIPartition method)

allreduce_data() (distdl.backends.mpi.partition.MPIPartition method) assemble_global_tensor_structure() (in module distdl.backends.mpi.tensor_comm)

B

backward() (distdl.backends.mpi.autograd.BroadcastFunction static method) (distdl.backends.mpi.autograd.DistributedTransposeFunction static method) (distdl.backends.mpi.autograd.HaloExchangeFunction static method) (distdl.backends.mpi.autograd.SumReduceFunction static method)

broadcast_data() (distdl.backends.mpi.partition.MPIPartition method) broadcast_tensor_structure() (in module distdl.backends.mpi.tensor_comm) BroadcastFunction (class in distdl.backends.mpi.autograd)

C

cartesian_index() (distdl.backends.mpi.partition.MPICartesianPartition method) CartesianPartition (in module distdl.backends.mpi) check_identical_comm() (in module distdl.backends.mpi.compare) check_identical_group() (in module distdl.backends.mpi.compare) check_null_comm() (in module distdl.backends.mpi.compare) check_null_group() (in module distdl.backends.mpi.compare) check_null_rank() (in module distdl.backends.mpi.compare)

ConvMixin (class in distdl.nn.mixins) create_broadcast_partition_to() (distdl.backends.mpi.partition.MPIPartition method) create_cartesian_subtopology_partition() (distdl.backends.mpi.partition.MPICartesianPartition method) create_cartesian_topology_partition() (distdl.backends.mpi.partition.MPIPartition method) create_partition_inclusive() (distdl.backends.mpi.partition.MPIPartition method) create_partition_union() (distdl.backends.mpi.partition.MPIPartition method) create_reduction_partition_to() (distdl.backends.mpi.partition.MPIPartition method)

D

deactivate() (distdl.backends.mpi.partition.MPICartesianPartition method) (distdl.backends.mpi.partition.MPIPartition method) distdl.backends.mpi.autograd.broadcast module distdl.backends.mpi.autograd.halo_exchange module distdl.backends.mpi.autograd.sum_reduce module distdl.backends.mpi.autograd.transpose module distdl.backends.mpi.compare module distdl.backends.mpi.partition module distdl.backends.mpi.tensor_comm module distdl.nn.halo_exchange module distdl.nn.mixins.halo_mixin module

distdl.nn.padnd module distdl.nn.unpadnd module DistributedChannelConv1d (class in distdl.nn.conv_channel) DistributedChannelConv2d (class in distdl.nn.conv_channel) DistributedChannelConv3d (class in distdl.nn.conv_channel) DistributedChannelConvBase (class in distdl.nn.conv_channel) DistributedConv1d (class in distdl.nn.conv) DistributedConv2d (class in distdl.nn.conv) DistributedConv3d (class in distdl.nn.conv) DistributedConvSelector (class in distdl.nn.conv) DistributedFeatureConv1d (class in distdl.nn.conv_feature) DistributedFeatureConv2d (class in distdl.nn.conv_feature) DistributedFeatureConv3d (class in distdl.nn.conv_feature) DistributedFeatureConvBase (class in distdl.nn.conv_feature) DistributedGeneralConv1d (class in distdl.nn.conv_general) DistributedGeneralConv2d (class in distdl.nn.conv_general) DistributedGeneralConv3d (class in distdl.nn.conv_general) DistributedGeneralConvBase (class in distdl.nn.conv_general) DistributedTransposeFunction (class in distdl.backends.mpi.autograd)

F

forward() (distdl.backends.mpi.autograd.BroadcastFunction static method) (distdl.backends.mpi.autograd.DistributedTransposeFunction static method) (distdl.backends.mpi.autograd.HaloExchangeFunction static method) (distdl.backends.mpi.autograd.SumReduceFunction static method) (distdl.nn.conv_channel.DistributedChannelConvBase method) (distdl.nn.conv_feature.DistributedFeatureConvBase method) (distdl.nn.conv_general.DistributedGeneralConvBase method)

H

HaloExchangeFunction (class in distdl.backends.mpi.autograd)

HaloMixin (class in distdl.nn.mixins)

M

module distdl.backends.mpi.autograd.broadcast distdl.backends.mpi.autograd.halo_exchange distdl.backends.mpi.autograd.sum_reduce distdl.backends.mpi.autograd.transpose distdl.backends.mpi.compare distdl.backends.mpi.partition distdl.backends.mpi.tensor_comm distdl.nn.halo_exchange distdl.nn.mixins.halo_mixin distdl.nn.padnd distdl.nn.unpadnd

Module (class in distdl.nn.module) MPICartesianPartition (class in distdl.backends.mpi.partition) MPIPartition (class in distdl.backends.mpi.partition)

N

neighbor_ranks() (distdl.backends.mpi.partition.MPICartesianPartition method)

P

PadNd (class in distdl.nn.padnd) PadNdFunction (class in distdl.nn.padnd)

Partition (in module distdl.backends.mpi) PoolingMixin (class in distdl.nn.mixins) print_sequential() (distdl.backends.mpi.partition.MPIPartition method)

S

SumReduceFunction (class in distdl.backends.mpi.autograd)

U

UnpadNd (class in distdl.nn.unpadnd)

UnpadNdFunction (class in distdl.nn.unpadnd)