gotensor

package

v0.0.0-...-87e57c7 Latest Latest Go to latest Published: Nov 3, 2023 License: MIT Imports: 3 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type GoTensor struct{}

func New() *GoTensor

func (*GoTensor) FP16Add(x, w, y []uint16)

func (*GoTensor) FP16Div(x, w, y []uint16)

func (*GoTensor) FP16Dot(x, w []uint16) uint16

func (*GoTensor) FP16Mul(x, w, y []uint16)

func (*GoTensor) FP16Pow(x []uint16, n uint16, y []uint16)

func (*GoTensor) FP16ScalarAdd(x uint16, w, y []uint16)

func (*GoTensor) FP16ScalarDiv(x uint16, w, y []uint16)

func (*GoTensor) FP16ScalarMul(x uint16, w, y []uint16)

func (*GoTensor) FP16ScalarSub(x uint16, w, y []uint16)

func (*GoTensor) FP16Sub(x, w, y []uint16)

func (*GoTensor) FP32Add(x, w, y []float32)

func (*GoTensor) FP32Div(x, w, y []float32)

func (*GoTensor) FP32Dot(x, w []float32) float32

func (*GoTensor) FP32Mul(x, w, y []float32)

func (*GoTensor) FP32Pow(x []float32, n float32, y []float32)

func (*GoTensor) FP32ScalarAdd(x float32, w, y []float32)

func (*GoTensor) FP32ScalarDiv(x float32, w, y []float32)

func (*GoTensor) FP32ScalarMul(x float32, w, y []float32)

func (*GoTensor) FP32ScalarSub(x float32, w, y []float32)

func (*GoTensor) FP32Sub(x, w, y []float32)