Module index_offset_with_layout 

Functions

expand

Returns the offset of the tensor corresponding to the layout tensor.

Type Aliases

N1

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

N2

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