TensorConvolutionOptions

import { TensorConvolutionOptions } from "destack:tensor";

Output options for tensor convolution.

export struct TensorConvolutionOptions<T, const Rank: usize, F: TensorFormat = Dense<MemoryOrder.RowMajor>, P: Placement = Unsharded> {
    groups: ConvolutionGroups;
    placement: P;
    device: Device;
    mesh: Mesh;
    stream: Stream;
    out: TensorView<T, Rank, F, P, "exclusive">;
}
language/library/src/tensor/options.ds:110:133

Members

groups: ConvolutionGroups

Convolution group counts.

placement: P

Result placement.

device: Device

Device used for unsharded placement.

mesh: Mesh

Mesh used for sharded placement.

stream: Stream

Stream used to order the work.

out: TensorView<T, Rank, F, P, "exclusive">

Exclusive output view to write into.