Module sparse_rows_multiply_kernel 

Structs

SparseRowsMultiplyKernel

sparse_rows_multiply_kernel Kernel

SparseRowsMultiplyKernelInfo

_F

Functions

expand

Spatial projection with at most width nonzero weights per output channel (ELLPACK rows): output[o, t] = Σ_k values[o, k] · input[indices[o, k], t]. Padding entries carry weight 0. One unit per (output channel, sample).

launch

Launch the kernel sparse_rows_multiply_kernel() on the given runtime

Type Aliases

F

Type to be used as a generic for launch kernel argument.