backends

package

Go to main page

Versions in this module

v0

v0.24.1

Oct 23, 2025

v0.24.1-cli-v0.1.2

Oct 23, 2025

v0.24.0

Oct 21, 2025

Changes in this version

+ var RngStateShape = shapes.Make(dtypes.Uint64, 3)

v0.24.0-cli-v0.1.1

Oct 21, 2025

v0.23.2

Sep 29, 2025

v0.23.1

Sep 25, 2025

v0.23.0

Sep 21, 2025

Changes in this version

type OpType

+ const OpTypeClamp

+ const OpTypeIsNaN

type StandardOps

+ BatchNormForInference func(operand, scale, offset, mean, variance Op, epsilon float32, featureAxis int) (Op, error)

+ BatchNormForTraining func(operand, scale, offset Op, epsilon float32, featureAxis int) (normalized Op, batchMean Op, batchVariance Op, err error)

+ BatchNormGradient func(operand, scale, mean, variance, gradOutput Op, epsilon float32, ...) (gradOperand Op, gradScale Op, gradOffset Op, err error)

+ BitCount func(operand Op) (Op, error)

+ Clamp func(min, x, max Op) (Op, error)

+ Identity func(x Op) (Op, error)

+ IsNaN func(x Op) (Op, error)

+ ReduceWindow func(x Op, reductionType ReduceOpType, ...) (Op, error)

+ RngBitGenerator func(state Op, shape shapes.Shape) (newState Op, values Op, err error)

v0.22.1

Aug 22, 2025

v0.22.0

Aug 22, 2025

Changes in this version

type ConvolveAxesConfig

+ InputChannels int

+ KernelInputChannels int

+ KernelOutputChannels int

+ OutputChannels int

type OpType

+ const OpTypeConvGeneral

type StandardOps

+ ConvGeneral func(input, kernel Op, axes ConvolveAxesConfig, strides []int, paddings [][2]int, ...) (Op, error)

v0.21.1

Aug 16, 2025

v0.21.0

Jul 1, 2025

v0.20.1

Jun 12, 2025

v0.20.0

Jun 3, 2025

Changes in this version

type Backend

+ func MustNew() Backend

+ func NewOrErr() (Backend, error)

+ IsFinalized func() bool

v0.19.5

May 30, 2025

v0.19.4

May 24, 2025

v0.19.3

May 20, 2025

Changes in this version

type Backend

+ String func() string

v0.19.2

May 1, 2025

v0.19.1

Apr 30, 2025

v0.19.0

Apr 29, 2025

Changes in this version

+ const ConfigEnvVar

+ func List() []string

+ func OpTypeStrings() []string

type Backend

+ Capabilities func() Capabilities

+ type Capabilities struct

+ DTypes map[dtypes.DType]bool

+ Operations map[OpType]bool

+ func (c Capabilities) Clone() Capabilities

+ type OpType int

+ const OpTypeAbs

+ const OpTypeAdd

+ const OpTypeArgMinMax

+ const OpTypeBatchNormForInference

+ const OpTypeBatchNormForTraining

+ const OpTypeBatchNormGradient

+ const OpTypeBitCount

+ const OpTypeBitcast

+ const OpTypeBitwiseAnd

+ const OpTypeBitwiseNot

+ const OpTypeBitwiseOr

+ const OpTypeBitwiseXor

+ const OpTypeBroadcast

+ const OpTypeBroadcastInDim

+ const OpTypeCeil

+ const OpTypeClz

+ const OpTypeComplex

+ const OpTypeConcatenate

+ const OpTypeConj

+ const OpTypeConstant

+ const OpTypeConvGeneralDilated

+ const OpTypeConvertDType

+ const OpTypeCos

+ const OpTypeDiv

+ const OpTypeDot

+ const OpTypeDotGeneral

+ const OpTypeDynamicSlice

+ const OpTypeDynamicUpdateSlice

+ const OpTypeEqual

+ const OpTypeEqualTotalOrder

+ const OpTypeErf

+ const OpTypeExp

+ const OpTypeExpm1

+ const OpTypeFFT

+ const OpTypeFloor

+ const OpTypeGather

+ const OpTypeGreaterOrEqual

+ const OpTypeGreaterOrEqualTotalOrder

+ const OpTypeGreaterThan

+ const OpTypeGreaterThanTotalOrder

+ const OpTypeIdentity

+ const OpTypeImag

+ const OpTypeInvalid

+ const OpTypeIota

+ const OpTypeIsFinite

+ const OpTypeLast

+ const OpTypeLessOrEqual

+ const OpTypeLessOrEqualTotalOrder

+ const OpTypeLessThan

+ const OpTypeLessThanTotalOrder

+ const OpTypeLog

+ const OpTypeLog1p

+ const OpTypeLogicalAnd

+ const OpTypeLogicalNot

+ const OpTypeLogicalOr

+ const OpTypeLogicalXor

+ const OpTypeLogistic

+ const OpTypeMax

+ const OpTypeMin

+ const OpTypeMul

+ const OpTypeNeg

+ const OpTypeNotEqual

+ const OpTypeNotEqualTotalOrder

+ const OpTypePad

+ const OpTypeParameter

+ const OpTypePow

+ const OpTypeReal

+ const OpTypeReduceBitwiseAnd

+ const OpTypeReduceBitwiseOr

+ const OpTypeReduceBitwiseXor

+ const OpTypeReduceLogicalAnd

+ const OpTypeReduceLogicalOr

+ const OpTypeReduceLogicalXor

+ const OpTypeReduceMax

+ const OpTypeReduceMin

+ const OpTypeReduceProduct

+ const OpTypeReduceSum

+ const OpTypeReduceWindow

+ const OpTypeRem

+ const OpTypeReshape

+ const OpTypeReverse

+ const OpTypeRngBitGenerator

+ const OpTypeRound

+ const OpTypeRsqrt

+ const OpTypeScatterMax

+ const OpTypeScatterMin

+ const OpTypeScatterSum

+ const OpTypeSelectAndScatterMax

+ const OpTypeSelectAndScatterMin

+ const OpTypeSelectAndScatterSum

+ const OpTypeShiftLeft

+ const OpTypeShiftRightArithmetic

+ const OpTypeShiftRightLogical

+ const OpTypeSign

+ const OpTypeSin

+ const OpTypeSlice

+ const OpTypeSqrt

+ const OpTypeSub

+ const OpTypeTanh

+ const OpTypeTranspose

+ const OpTypeWhere

+ func OpTypeString(s string) (OpType, error)

+ func OpTypeValues() []OpType

+ func (i OpType) IsAOpType() bool

+ func (i OpType) String() string

v0.18.1

Apr 13, 2025

v0.18.0

Apr 11, 2025

Changes in this version

type StandardOps

+ ScatterSum func(operand, scatterIndices, updates Op, indexVectorAxis int, ...) Op

v0.17.1

Feb 26, 2025

Changes in this version

+ func FFTTypeStrings() []string

+ func ReduceOpTypeStrings() []string

type FFTType

+ func FFTTypeString(s string) (FFTType, error)

+ func FFTTypeValues() []FFTType

+ func (i FFTType) IsAFFTType() bool

type ReduceOpType

+ func ReduceOpTypeString(s string) (ReduceOpType, error)

+ func ReduceOpTypeValues() []ReduceOpType

+ func (i ReduceOpType) IsAReduceOpType() bool

type StandardOps

+ Bitcast func(x Op, targetDType dtypes.DType) Op

v0.17.0

Feb 1, 2025

Changes in this version

type StandardOps

+ BitwiseAnd func(x0, x1 Op) Op

+ BitwiseNot func(x Op) Op

+ BitwiseOr func(x0, x1 Op) Op

+ BitwiseXor func(x0, x1 Op) Op

+ LogicalAnd func(x0, x1 Op) Op

+ LogicalOr func(x0, x1 Op) Op

+ LogicalXor func(x0, x1 Op) Op

+ ReduceBitwiseAnd func(x Op, axes ...int) Op

+ ReduceBitwiseOr func(x Op, axes ...int) Op

+ ReduceBitwiseXor func(x Op, axes ...int) Op

+ ReduceLogicalAnd func(x Op, axes ...int) Op

+ ReduceLogicalOr func(x Op, axes ...int) Op

+ ReduceLogicalXor func(x Op, axes ...int) Op

+ ShiftLeft func(x0, x1 Op) Op

+ ShiftRightArithmetic func(x0, x1 Op) Op

+ ShiftRightLogical func(x0, x1 Op) Op

v0.16.1

Dec 19, 2024

v0.16.0

Dec 19, 2024

Changes in this version

type DataInterface

+ BufferData func(buffer Buffer) (flat any)

+ HasSharedBuffers func() bool

+ NewSharedBuffer func(deviceNum DeviceNum, shape shapes.Shape) (buffer Buffer, flat any)

v0.15.3

Nov 25, 2024

v0.15.2

Nov 17, 2024

v0.15.1

Nov 11, 2024

v0.15.0

Nov 1, 2024

v0.14.0

Oct 24, 2024

v0.13.0

Oct 7, 2024

Changes in this version

type Builder

+ BitCount func(operand Op) Op

type StandardOps

+ IsFinite func(x Op) Op

v0.12.0

Sep 23, 2024

Changes in this version

type StandardOps

+ Erf func(x Op) Op

v0.11.3

Aug 29, 2024

v0.11.2

Aug 28, 2024

v0.11.1

Aug 28, 2024

Changes in this version

type StandardOps

+ DynamicSlice func(operand Op, startIndices []Op, sliceDims []int) Op

+ DynamicUpdateSlice func(operand, update Op, startIndices []Op) Op

+ ReduceAnd func(x Op, axes ...int) Op

+ ReduceOr func(x Op, axes ...int) Op

v0.11.0

Aug 16, 2024

Changes in this version

+ const GOMLX_BACKEND

+ var DefaultConfig string

+ func NotImplemented()

+ func Register(name string, constructor Constructor)

+ type Backend interface

+ Builder func(name string) Builder

+ Description func() string

+ Finalize func()

+ Name func() string

+ NumDevices func() DeviceNum

+ func New() Backend

+ func NewWithConfig(config string) Backend

+ type Buffer any

+ type Builder interface

+ BatchNormForInference func(operand, scale, offset, mean, variance Op, epsilon float32, axis int) Op

+ BatchNormForTraining func(operand, scale, offset Op, epsilon float32, axis int) (normalized, batchMean, batchVariance Op)

+ BatchNormGradient func(operand, scale, mean, variance, gradOutput Op, epsilon float32, axis int) (gradOperand, gradScale, gradOffset Op)

+ Compile func(outputs ...Op) Executable

+ Constant func(flat any, dims ...int) Op

+ Identity func(x Op) Op

+ Name func() string

+ OpShape func(op Op) shapes.Shape

+ Parameter func(name string, shape shapes.Shape) Op

+ ReduceWindow func(x Op, reductionType ReduceOpType, ...) Op

+ RngBitGenerator func(state Op, shape shapes.Shape) (newState, values Op)

+ type Constructor func(config string) Backend

+ type ConvolveAxesConfig struct

+ InputBatch int

+ InputChannel int

+ InputSpatial []int

+ KernelInputChannel int

+ KernelOutputChannel int

+ KernelSpatial []int

+ OutputBatch int

+ OutputChannel int

+ OutputSpatial []int

+ func (c ConvolveAxesConfig) Clone() ConvolveAxesConfig

+ type DataInterface interface

+ BufferDeviceNum func(buffer Buffer) DeviceNum

+ BufferFinalize func(buffer Buffer)

+ BufferFromFlatData func(deviceNum DeviceNum, flat any, shape shapes.Shape) Buffer

+ BufferShape func(buffer Buffer) shapes.Shape

+ BufferToFlatData func(buffer Buffer, flat any)

+ type DeviceNum int

+ type Executable interface

+ Execute func(inputs []Buffer, donate []bool) []Buffer

+ Finalize func()

+ Inputs func() (names []string, inputShapes []shapes.Shape)

+ Outputs func() (outputShapes []shapes.Shape)

+ type FFTType int

+ const FFTForward

+ const FFTForwardReal

+ const FFTInverse

+ const FFTInverseReal

+ func (i FFTType) String() string

+ type Op any

+ type PadAxis struct

+ End int

+ Interior int

+ Start int

+ type ReduceOpType int

+ const ReduceOpMax

+ const ReduceOpMin

+ const ReduceOpProduct

+ const ReduceOpSum

+ const ReduceOpUndefined

+ func (i ReduceOpType) String() string

+ type StandardOps interface

+ Abs func(x Op) Op

+ Add func(x0, x1 Op) Op

+ And func(x0, x1 Op) Op

+ ArgMinMax func(x Op, axis int, outputDType dtypes.DType, isMin bool) Op

+ Broadcast func(x Op, prefixDims ...int) Op

+ BroadcastInDim func(x Op, outputShape shapes.Shape, broadcastAxes []int) Op

+ Ceil func(x Op) Op

+ Clz func(x Op) Op

+ Complex func(x0, x1 Op) Op

+ Concatenate func(axis int, operands ...Op) Op

+ Conj func(x Op) Op

+ ConvGeneralDilated func(operand, filter Op, axes ConvolveAxesConfig, strides []int, paddings [][2]int, ...) Op

+ ConvertDType func(x Op, dtype dtypes.DType) Op

+ Cos func(x Op) Op

+ Div func(x0, x1 Op) Op

+ Dot func(x0, x1 Op) Op

+ DotGeneral func(lhs Op, lhsContractingAxes, lhsBatchAxes []int, rhs Op, ...) Op

+ Equal func(x0, x1 Op) Op

+ EqualTotalOrder func(x0, x1 Op) Op

+ Exp func(x Op) Op

+ Expm1 func(x Op) Op

+ FFT func(operand Op, fftType FFTType, fftLength []int) Op

+ Floor func(x Op) Op

+ Gather func(operand, startIndices Op, indexVectorAxis int, ...) Op

+ GreaterOrEqual func(x0, x1 Op) Op

+ GreaterOrEqualTotalOrder func(x0, x1 Op) Op

+ GreaterThan func(x0, x1 Op) Op

+ GreaterThanTotalOrder func(x0, x1 Op) Op

+ Imag func(x Op) Op

+ Iota func(shape shapes.Shape, iotaAxis int) Op

+ LessOrEqual func(x0, x1 Op) Op

+ LessOrEqualTotalOrder func(x0, x1 Op) Op

+ LessThan func(x0, x1 Op) Op

+ LessThanTotalOrder func(x0, x1 Op) Op

+ Log func(x Op) Op

+ Log1p func(x Op) Op

+ LogicalNot func(x Op) Op

+ Logistic func(x Op) Op

+ Max func(x0, x1 Op) Op

+ Min func(x0, x1 Op) Op

+ Mul func(x0, x1 Op) Op

+ Neg func(x Op) Op

+ NotEqual func(x0, x1 Op) Op

+ NotEqualTotalOrder func(x0, x1 Op) Op

+ Or func(x0, x1 Op) Op

+ Pad func(x, fillValue Op, axesConfig ...PadAxis) Op

+ Pow func(x0, x1 Op) Op

+ Real func(x Op) Op

+ ReduceMax func(x Op, axes ...int) Op

+ ReduceMin func(x Op, axes ...int) Op

+ ReduceProduct func(x Op, axes ...int) Op

+ ReduceSum func(x Op, axes ...int) Op

+ Rem func(x0, x1 Op) Op

+ Reshape func(x Op, dimensions ...int) Op

+ Reverse func(x Op, axes ...int) Op

+ Round func(x Op) Op

+ Rsqrt func(x Op) Op

+ ScatterAdd func(operand, scatterIndices, updates Op, indexVectorAxis int, ...) Op

+ ScatterMax func(operand, scatterIndices, updates Op, indexVectorAxis int, ...) Op

+ ScatterMin func(operand, scatterIndices, updates Op, indexVectorAxis int, ...) Op

+ SelectAndScatterMax func(operand, source Op, windowDimensions, windowStrides []int, paddings [][2]int) Op

+ SelectAndScatterMin func(operand, source Op, windowDimensions, windowStrides []int, paddings [][2]int) Op

+ SelectAndScatterSum func(operand, source Op, windowDimensions, windowStrides []int, paddings [][2]int) Op

+ Sign func(x Op) Op

+ Sin func(x Op) Op

+ Slice func(x Op, starts, limits, strides []int) Op

+ Sqrt func(x Op) Op

+ Sub func(x0, x1 Op) Op

+ Tanh func(x Op) Op

+ Transpose func(x Op, permutations ...int) Op

+ Where func(condition, onTrue, onFalse Op) Op

+ Xor func(x0, x1 Op) Op

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL