For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo struct

StaticTensorSpecList

struct StaticTensorSpecList[dtype: DType, rank: Int, //, internals_list: ParameterList[internals_list.values], shapes_list: ParameterList[shapes_list.values], strides_list: ParameterList[strides_list.values]]

A statically indexable list of data that can be assembled into a StaticTensorSpecList on demand. This handles the complexities that arise with heterogenous specs.

Implemented traits

AnyType, Deinitable, Movable

Methods

__getitem_param__

def __getitem_param__[index: Int](self, out result: StaticTensorSpec[dtype, rank, Layout[TypeList[#kgen.param_list.tabulate(rank, [idx: __mlir_type.index] ComptimeInt[shapes_list.values[SIMDLength(index)][Int(idx)]] if (xor identical((xor (lt shapes_list.values[SIMDLength(index)][Int(idx)], 0), True), False), True) else Int)](), TypeList[#kgen.param_list.tabulate(rank, [idx: __mlir_type.index] ComptimeInt[strides_list.values[SIMDLength(index)][Int(idx)]] if (xor identical((xor (lt strides_list.values[SIMDLength(index)][Int(idx)], 0), True), False), True) else Int)]()]])

Returns:

StaticTensorSpec[dtype, rank, Layout[TypeList[#kgen.param_list.tabulate(rank, [idx: __mlir_type.index] ComptimeInt[shapes_list.values[SIMDLength(index)][Int(idx)]] if (xor identical((xor (lt shapes_list.values[SIMDLength(index)][Int(idx)], 0), True), False), True) else Int)](), TypeList[#kgen.param_list.tabulate(rank, [idx: __mlir_type.index] ComptimeInt[strides_list.values[SIMDLength(index)][Int(idx)]] if (xor identical((xor (lt strides_list.values[SIMDLength(index)][Int(idx)], 0), True), False), True) else Int)]()]]