Skip to main content

Module distributed_ops

Module distributed_ops 

Source
Expand description

Distributed Operations for Tensor Computations

High-level distributed tensor operations that use the MeshComm trait. These operations enable distributed inference across multiple devices.

Structs§

DistributedOps
Distributed tensor operations

Functions§

dist_ops
Convenience function to create distributed ops from a mesh

Type Aliases§

DistResult
Result type for distributed operations