Skip to main content

Module element

ruda_tensor_device

Module element 

Source
Expand description

Scalar element contracts for device tensor operations.

Traitsยง

BoolElement
The element type for booleans for the jit backend.
FloatElement
The float element type for the jit backend.
IntElement
The int element type for the jit backend.
MatmulElement
Element that can be used for matrix multiplication. Includes ints and floats.
TensorElement
The base element trait for the jit backend.