BasaltBasalt

attributes

Aliases

MAX_ATTRS = 10MAX_NAME_CHARS = 16MAX_DATA_BYTES = 32

Structs

AttributeVector
  • Fields
    • attributes - StaticTuple[Attribute, 10]
    • size - Int
  • Functions
Attribute
  • Fields
    • name - Bytes[16]
    • data - Bytes[32]
    • data_shape - StaticIntTuple[8]
  • Functions

graph

Structs

Graph
  • Fields
    • inputs - List[Symbol]
    • params - ParamDict
    • nodes - List[Node]
    • outputs - List[Symbol]
    • loss_out - Optional[Symbol]
    • symbol_count - SIMD[ui32, 1]
  • Functions

node

Structs

Node
  • Fields
    • operator - OP
    • output - Symbol
    • input_1 - Symbol
    • input_2 - Optional[Symbol]
    • input_3 - Optional[Symbol]
    • attributes - AttributeVector
  • Functions

params

Structs

Param
  • Fields
    • data - Optional[List[SIMD[f32, 1]]]
    • initializer - Optional[Attribute]
  • Functions
ParamDict
  • Fields
    • symbols - List[Symbol]
    • values - List[Param]
  • Functions

symbol

Structs

Symbol
  • Fields
    • name - SIMD[ui32, 1]
    • dtype - DType
    • shape - TensorShape
    • trainable - Bool
  • Functions