xlabuilder

package

Go to main page

Versions in this module

v0

v0.6.2

Feb 26, 2025

v0.6.1

Feb 26, 2025

Changes in this version

+ func DecodeBitcast(op *Op) (dtype dtypes.DType)

type Op

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

type OpType

+ const BitcastOp

v0.6.0

Feb 1, 2025

Changes in this version

+ var MatchingCVersion = "v0.6.0"

+ func CVersion() string

type Op

+ func BitwiseAnd(x0, x1 *Op) (*Op, error)

+ func BitwiseNot(x *Op) (*Op, error)

+ func BitwiseOr(x0, x1 *Op) (*Op, error)

+ func BitwiseXor(x0, x1 *Op) (*Op, error)

+ func LogicalAnd(x0, x1 *Op) (*Op, error)

+ func LogicalOr(x0, x1 *Op) (*Op, error)

+ func LogicalXor(x0, x1 *Op) (*Op, error)

+ func Not(operand *Op) (*Op, error)

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

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

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

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

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

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

+ func ReduceXor(x *Op, axes ...int) (*Op, error)

type OpType

+ const BitwiseAndOp

+ const BitwiseNotOp

+ const BitwiseOrOp

+ const BitwiseXorOp

+ const LogicalAndOp

+ const LogicalOrOp

+ const LogicalXorOp

type ReduceOpType

+ const ReduceBitwiseAndType

+ const ReduceBitwiseOrType

+ const ReduceBitwiseXorType

+ const ReduceLogicalAndType

+ const ReduceLogicalOrType

+ const ReduceLogicalXorType

v0.5.1

Jan 29, 2025

Changes in this version

type Op

+ func ShiftLeft(x0, x1 *Op) (*Op, error)

+ func ShiftRightArithmetic(x0, x1 *Op) (*Op, error)

+ func ShiftRightLogical(x0, x1 *Op) (*Op, error)

type OpType

+ const ShiftLeftOp

+ const ShiftRightArithmeticOp

+ const ShiftRightLogicalOp

v0.5.0

Dec 19, 2024

v0.4.9

Nov 25, 2024

v0.4.8

Nov 19, 2024

v0.4.7

Nov 17, 2024

v0.4.6

Nov 12, 2024

Changes in this version

+ var RngStateShape = MakeShape(dtypes.U64, 3)

+ type RngAlgorithm int

+ const RngAlgorithmDefault

+ const RngAlgorithmPhilox

+ const RngAlgorithmThreeFry

type Shape

+ func (s Shape) Equal(s2 Shape) bool

v0.4.5

Nov 6, 2024

v0.4.4

Oct 24, 2024

v0.4.3

Oct 23, 2024

Changes in this version

+ func HasStableHLO() bool

type Literal

+ func (l *Literal) Data(accessFn func(data []byte))

type XlaComputation

+ func (comp *XlaComputation) HasStableHLO() bool

+ func (comp *XlaComputation) SerializedStableHLO() (*cbuffer.CBuffer, error)

+ func (comp *XlaComputation) TextStableHLO() (string, error)

v0.4.2

Oct 3, 2024

Changes in this version

type Op

+ func IsFinite(x *Op) (*Op, error)

+ func PopulationCount(x *Op) (*Op, error)

type OpType

+ const IsFiniteOp

+ const PopulationCountOp

v0.4.1

Sep 28, 2024

v0.4.0

Sep 23, 2024

Changes in this version

type Op

+ func Erf(x *Op) (*Op, error)

type OpType

+ const ErfOp

v0.3.2

Aug 28, 2024

Changes in this version

type Op

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

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

type ReduceOpType

+ const ReduceAndType

+ const ReduceOrType

v0.3.1

Aug 27, 2024

v0.3.0

Aug 23, 2024

Changes in this version

type Shape

+ func MakeShapeOrError(dtype dtypes.DType, dimensions ...int) (Shape, error)

v0.2.4

Aug 20, 2024

v0.2.3

Aug 16, 2024

v0.2.2

Aug 16, 2024

v0.2.1

Aug 11, 2024

Changes in this version

+ func DecodeDynamicSlice(op *Op) (operand *Op, startIndices []*Op, sliceDims []int)

+ func DecodeDynamicUpdateSlice(op *Op) (operand, update *Op, startIndices []*Op)

type Op

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

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

type OpType

+ const DynamicSliceOp

+ const DynamicUpdateSliceOp

v0.2.0

Jul 18, 2024

Changes in this version

type Literal

+ func NewArrayLiteralFromAny(flatAny any, dimensions ...int) *Literal

type Op

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

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

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

+ func DecodeBatchNormForInference(op *Op) (operand, scale, offset, mean, variance *Op, epsilon float32, axis int)

+ func DecodeBatchNormForTraining(op *Op) (operand, scale, offset *Op, epsilon float32, axis int)

+ func (op *Op) Builder() *XlaBuilder

type Shape

+ func (s Shape) Memory() uintptr

v0.1.2

Jul 5, 2024

v0.1.1

Jul 4, 2024

Changes in this version

+ func DecodeWhile(op *Op) (initialState *Op, condition, body *XlaComputation)

type Op

+ func While(initialState *Op, condition, body *XlaComputation) (*Op, error)

type OpType

+ const WhileOp

type Shape

+ func (s Shape) Clone() (newS Shape)

v0.0.1

Jul 3, 2024

Changes in this version

+ func DecodeArgMinMax(op *Op) (axis int, outputDType dtypes.DType, isMin bool)

+ func DecodeBroadcast(op *Op) (prefixDims []int)

+ func DecodeConcatenate(op *Op) (axis int)

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

+ func DecodeConvertDType(op *Op) (dtype dtypes.DType)

+ func DecodeDotGeneral(op *Op) (lhs *Op, lhsContractingAxes, lhsBatchAxes []int, rhs *Op, ...)

+ func DecodeGather(op *Op) (indexVectorAxis int, ...)

+ func DecodeGetTupleElement(op *Op) (elementIdx int)

+ func DecodeReduceWindow(op *Op) (reduceType ReduceOpType, reduceComputation *XlaComputation, initialValue *Op, ...)

+ func DecodeReshape(op *Op) (dimensions []int)

+ func DecodeRngBitGenerator(op *Op) (state *Op, shape Shape)

+ func DecodeScatter(op *Op) (indexVectorAxis int, ...)

+ func DecodeSelectAndScatter(op *Op) (operand, source, defaultValue *Op, ...)

+ func DecodeSlice(op *Op) (starts, limits, strides []int)

+ func DecodeTranspose(op *Op) (permutations []int)

+ type ConvolveAxesConfig struct

+ InputBatch int

+ InputChannel int

+ InputSpatial []int

+ KernelInputChannel int

+ KernelOutputChannel int

+ KernelSpatial []int

+ OutputBatch int

+ OutputChannel int

+ OutputSpatial []int

+ type Literal struct

+ func NewArrayLiteral[T dtypes.Supported](flat []T, dimensions ...int) *Literal

+ func NewLiteralFromShape(shape Shape) *Literal

+ func NewScalarLiteralFromAny(value any) *Literal

+ func NewScalarLiteralFromFloat64(value float64, dtype dtypes.DType) *Literal

+ func NewScalarLiteral[T dtypes.Supported](value T) *Literal

+ func (l *Literal) Destroy()

+ func (l *Literal) IsNil() bool

+ func (l *Literal) Shape() Shape

+ type Op struct

+ ComputationArg *XlaComputation

+ FloatArg float32

+ IntArg int

+ IntsArg []int

+ LiteralArg *Literal

+ OpInputs []*Op

+ ReduceType ReduceOpType

+ SecondComputationArg *XlaComputation

+ Shape Shape

+ ShapeArg Shape

+ StrArg string

+ Type OpType

+ UserPayload any

+ func Abs(x *Op) (*Op, error)

+ func Add(x0, x1 *Op) (*Op, error)

+ func And(x0, x1 *Op) (*Op, error)

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

+ func BatchNormGrad(operand, scale, mean, variance, gradOutput *Op, epsilon float32, axis int) (gradOperand, gradScale, gradOffset *Op, err error)

+ func BatchNormInference(operand, scale, offset, mean, variance *Op, epsilon float32, axis int) (*Op, error)

+ func BatchNormTraining(operand, scale, offset *Op, epsilon float32, axis int) (normalized, batchMean, batchVariance *Op, err error)

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

+ func BroadcastInDim(x *Op, outputShape Shape, broadcastAxes []int) (*Op, error)

+ func Call(builder *XlaBuilder, subComputation *XlaComputation, operands ...*Op) (*Op, error)

+ func Ceil(x *Op) (*Op, error)

+ func Clz(x *Op) (*Op, error)

+ func Complex(x0, x1 *Op) (*Op, error)

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

+ func Conj(x *Op) (*Op, error)

+ func Constant(builder *XlaBuilder, x *Literal) (*Op, error)

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

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

+ func Cos(x *Op) (*Op, error)

+ func DecodeBatchNormGrad(op *Op) (operand, scale, mean, variance, gradOutput *Op, epsilon float32, axis int)

+ func DecodeBatchNormInference(op *Op) (operand, scale, offset, mean, variance *Op, epsilon float32, axis int)

+ func DecodeBatchNormTraining(op *Op) (operand, scale, offset *Op, epsilon float32, axis int)

+ func DecodeFFT(op *Op) (operand *Op, fftType protos.FftType, fftLength []int)

+ func DecodeReverse(op *Op) (x *Op, axes []int)

+ func Div(x0, x1 *Op) (*Op, error)

+ func Dot(x0, x1 *Op) (*Op, error)

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

+ func Equal(x0, x1 *Op) (*Op, error)

+ func EqualTotalOrder(x0, x1 *Op) (*Op, error)

+ func Exp(x *Op) (*Op, error)

+ func Expm1(x *Op) (*Op, error)

+ func FFT(operand *Op, fftType protos.FftType, fftLength []int) (*Op, error)

+ func Floor(x *Op) (*Op, error)

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

+ func GetTupleElement(input *Op, elementIdx int) (*Op, error)

+ func GreaterOrEqual(x0, x1 *Op) (*Op, error)

+ func GreaterOrEqualTotalOrder(x0, x1 *Op) (*Op, error)

+ func GreaterThan(x0, x1 *Op) (*Op, error)

+ func GreaterThanTotalOrder(x0, x1 *Op) (*Op, error)

+ func Identity(input *Op) *Op

+ func Imag(x *Op) (*Op, error)

+ func Iota(builder *XlaBuilder, shape Shape, iotaAxis int) (*Op, error)

+ func LessOrEqual(x0, x1 *Op) (*Op, error)

+ func LessOrEqualTotalOrder(x0, x1 *Op) (*Op, error)

+ func LessThan(x0, x1 *Op) (*Op, error)

+ func LessThanTotalOrder(x0, x1 *Op) (*Op, error)

+ func Log(x *Op) (*Op, error)

+ func Log1p(x *Op) (*Op, error)

+ func LogicalNot(x *Op) (*Op, error)

+ func Logistic(x *Op) (*Op, error)

+ func Max(x0, x1 *Op) (*Op, error)

+ func Min(x0, x1 *Op) (*Op, error)

+ func Mul(x0, x1 *Op) (*Op, error)

+ func Neg(x *Op) (*Op, error)

+ func NotEqual(x0, x1 *Op) (*Op, error)

+ func NotEqualTotalOrder(x0, x1 *Op) (*Op, error)

+ func Or(x0, x1 *Op) (*Op, error)

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

+ func Parameter(builder *XlaBuilder, name string, paramIndex int, shape Shape) (*Op, error)

+ func Pow(x0, x1 *Op) (*Op, error)

+ func Real(x *Op) (*Op, error)

+ func Reduce(x *Op, reduceComputation *XlaComputation, initialValue *Op, axes ...int) (*Op, error)

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

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

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

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

+ func Rem(x0, x1 *Op) (*Op, error)

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

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

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

+ func Round(x *Op) (*Op, error)

+ func Rsqrt(x *Op) (*Op, error)

+ func ScalarOne(builder *XlaBuilder, dtype dtypes.DType) (*Op, error)

+ func ScalarZero(builder *XlaBuilder, dtype dtypes.DType) (*Op, error)

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

+ func ScatterCustom(operand, scatterIndices, updates *Op, updateComputation *XlaComputation, ...) (*Op, error)

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

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

+ func SelectAndScatterCustom(operand, source, defaultValue *Op, ...) (*Op, error)

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

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

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

+ func Sign(x *Op) (*Op, error)

+ func Sin(x *Op) (*Op, error)

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

+ func SplitTuple(tuple *Op) ([]*Op, error)

+ func Sqrt(x *Op) (*Op, error)

+ func Sub(x0, x1 *Op) (*Op, error)

+ func Tanh(x *Op) (*Op, error)

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

+ func Tuple(inputs ...*Op) (*Op, error)

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

+ func Xor(x0, x1 *Op) (*Op, error)

+ type OpType int32

+ const AbsOp

+ const AddOp

+ const AndOp

+ const ArgMinMaxOp

+ const BatchNormGradOp

+ const BatchNormInferenceOp

+ const BatchNormTrainingOp

+ const BroadcastInDimOp

+ const BroadcastOp

+ const CallOp

+ const CeilOp

+ const ClzOp

+ const ComplexOp

+ const ConcatenateOp

+ const ConjOp

+ const ConstantOp

+ const ConvGeneralDilatedOp

+ const ConvertDTypeOp

+ const CosOp

+ const DivOp

+ const DotGeneralOp

+ const DotOp

+ const EqualOp

+ const EqualTotalOrderOp

+ const ExpOp

+ const Expm1Op

+ const FftOp

+ const FloorOp

+ const GatherOp

+ const GetTupleElementOp

+ const GreaterOrEqualOp

+ const GreaterOrEqualTotalOrderOp

+ const GreaterThanOp

+ const GreaterThanTotalOrderOp

+ const IdentityOp

+ const ImagOp

+ const InvalidOp

+ const IotaOp

+ const LessOrEqualOp

+ const LessOrEqualTotalOrderOp

+ const LessThanOp

+ const LessThanTotalOrderOp

+ const Log1pOp

+ const LogOp

+ const LogicalNotOp

+ const LogisticOp

+ const MaxOp

+ const MinOp

+ const MulOp

+ const NegOp

+ const NotEqualOp

+ const NotEqualTotalOrderOp

+ const OrOp

+ const PadOp

+ const ParameterOp

+ const PowOp

+ const RealOp

+ const ReduceOp

+ const ReduceWindowOp

+ const RemOp

+ const ReshapeOp

+ const ReverseOp

+ const RngBitGeneratorOp

+ const RoundOp

+ const RsqrtOp

+ const ScatterOp

+ const SelectAndScatterOp

+ const SignOp

+ const SinOp

+ const SliceOp

+ const SqrtOp

+ const SubOp

+ const TanhOp

+ const TransposeOp

+ const TupleOp

+ const WhereOp

+ const XorOp

+ func (i OpType) String() string

+ type PadAxis struct

+ End int

+ Interior int

+ Start int

+ func DecodePad(op *Op) (axesConfig []PadAxis)

+ type ReduceOpType int

+ const ReduceMaxType

+ const ReduceMinType

+ const ReduceProductType

+ const ReduceSumType

+ const UndefinedReduceType

+ func (i ReduceOpType) String() string

+ type ReduceWindowConfig struct

+ func ReduceWindow(x *Op, windowDimensions []int) *ReduceWindowConfig

+ func (r *ReduceWindowConfig) Done() (*Op, error)

+ func (r *ReduceWindowConfig) Max() *ReduceWindowConfig

+ func (r *ReduceWindowConfig) Min() *ReduceWindowConfig

+ func (r *ReduceWindowConfig) Product() *ReduceWindowConfig

+ func (r *ReduceWindowConfig) Sum() *ReduceWindowConfig

+ func (r *ReduceWindowConfig) UseComputation(reduceComputation *XlaComputation, initialValue *Op) *ReduceWindowConfig

+ func (r *ReduceWindowConfig) WithBaseDilations(baseDilations []int) *ReduceWindowConfig

+ func (r *ReduceWindowConfig) WithPadding(paddings [][2]int) *ReduceWindowConfig

+ func (r *ReduceWindowConfig) WithStrides(strides []int) *ReduceWindowConfig

+ func (r *ReduceWindowConfig) WithWindowDilations(windowDilations []int) *ReduceWindowConfig

+ type Shape struct

+ DType dtypes.DType

+ Dimensions []int

+ TupleShapes []Shape

+ func DecodeBroadcastInDim(op *Op) (outputShape Shape, broadcastAxes []int)

+ func DecodeIota(op *Op) (shape Shape, iotaAxis int)

+ func DecodeParameter(paramOp *Op) (name string, paramIndex int, shape Shape)

+ func MakeShape(dtype dtypes.DType, dimensions ...int) Shape

+ func (s Shape) IsScalar() bool

+ func (s Shape) Rank() int

+ func (s Shape) Size() int

+ func (s Shape) String() string

+ func (s Shape) TupleSize() int

+ type XlaBuilder struct

+ func New(name string) *XlaBuilder

+ func (b *XlaBuilder) Build(outputOp *Op) (*XlaComputation, error)

+ func (b *XlaBuilder) CreateSubBuilder(computationName string) *XlaBuilder

+ func (b *XlaBuilder) Destroy()

+ func (b *XlaBuilder) GetReduceComputationAndInitialValue(reduction ReduceOpType, dtype dtypes.DType) (comp *XlaComputation, initialValue *Op, err error)

+ func (b *XlaBuilder) GetSelectAndScatterComputation(reduction ReduceOpType, dtype dtypes.DType) (selectComputation, scatterComputation *XlaComputation, err error)

+ func (b *XlaBuilder) IsNil() bool

+ func (b *XlaBuilder) Name() string

+ type XlaComputation struct

+ func (comp *XlaComputation) Destroy()

+ func (comp *XlaComputation) IsNil() bool

+ func (comp *XlaComputation) Name() string

+ func (comp *XlaComputation) SerializedHLO() *cbuffer.CBuffer

+ func (comp *XlaComputation) TextHLO() string

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