Functions — github.com/adalkiran/llama-nuts-and-bolts

hub / github.com/adalkiran/llama-nuts-and-bolts / functions

Functions352 in github.com/adalkiran/llama-nuts-and-bolts

⨍ Functions 352◇ Types & classes 51

↓ 131 callersMethodError

()

src/pickle/types.go:51

↓ 74 callersFunctionBFloat16fromFloat32

(f32 float32)

src/dtype/bfloat16.go:31

↓ 51 callersMethodDebugPrintf

(format string, v ...any)

src/common/logging.go:47

↓ 50 callersFunctionCompareTestTensor

(expected interface{}, expectedSize []int, actual *Tensor, floatThreshold float64, shorten bool)

src/ml/utils.go:115

↓ 35 callersMethodSetItem

(loc []int, val any)

src/ml/tensor.go:166

↓ 33 callersFunctionNewEmptyTensor

(size []int, dataType DataType)

src/ml/tensor.go:45

↓ 28 callersFunctionCompareTestTensorSkippable

(skip bool, expected interface{}, expectedSize []int, actual *Tensor, floatThreshold float64, shorten bool)

src/ml/utils.go:108

↓ 22 callersMethodcalculateByteOffset

(loc []int)

src/ml/tensor.go:245

↓ 20 callersMethodHasNext

()

src/ml/tensoriterators.go:30

↓ 20 callersMethodNext

()

src/ml/tensoriterators.go:13

↓ 18 callersFunctionLinearTransformation

(input *Tensor, weights *Tensor)

src/ml/operations_impl.go:427

↓ 17 callersMethodReshape

(newSize []int)

src/ml/tensor.go:386

↓ 15 callersFunctionARange

(start int, end int, step int, dataType DataType)

src/ml/operations_impl.go:11

↓ 14 callersMethodAppend

(item interface{})

src/pickle/picklereader.go:70

↓ 14 callersMethodFloat32

()

src/dtype/bfloat16.go:19

↓ 14 callersMethodSlice

(locStart []int, locEnd []int)

src/ml/tensor.go:266

↓ 13 callersMethodGetItem

(loc []int)

src/ml/tensor.go:154

↓ 13 callersMethodTranspose

(dim1 int, dim2 int)

src/ml/tensor.go:542

↓ 11 callersFunctionFull

(size []int, dataType DataType, fillValue any)

src/ml/operations_impl.go:55

↓ 11 callersMethodGetItemByOffset_AsFloat32

(offset int)

src/ml/tensor.go:181

↓ 11 callersFunctionIterateOver

(tensor *Tensor, ignoreTrailingDimensions int)

src/ml/tensoriterators.go:93

↓ 11 callersFunctionTriangularUpper

(input *Tensor, diagonal int)

src/ml/operations_impl.go:175

↓ 10 callersMethodConsolePrintf

(format string, v ...any)

src/common/logging.go:35

↓ 9 callersMethodForward

(infContext *InferenceContext, x *ml.Tensor)

src/model/llamatransformer.go:633

↓ 9 callersFunctiongetLayerTensor

(model *Model, nameFormat string, layerIndex int, expectedShape []int)

src/model/loader.go:194

↓ 9 callersMethodprintLinef

(format string, v ...any)

cmd/main.go:452

↓ 8 callersFunctionCheckBroadcastableOnce

(size1 []int, size2 []int)

src/ml/tensor.go:399

↓ 8 callersMethodRead

(byteCount int)

src/pickle/picklereader.go:46

↓ 8 callersFunctioncreateTestInputTensor

(size []int)

src/ml/operations_test.go:52

↓ 7 callersMethodGetBytesCount

()

src/ml/tensor.go:67

↓ 7 callersFunctionMean

(input *Tensor, dim int, keepdim bool)

src/ml/operations_impl.go:219

↓ 7 callersFunctionMultiplyElementwise

(input *Tensor, other *Tensor)

src/ml/operations_impl.go:367

↓ 7 callersFunctionNewEmptyTensorEx

(name string, size []int, dataType DataType, allocateRawData bool)

src/ml/tensor.go:32

↓ 7 callersMethodSetSlice

(locStart []int, locEnd []int, val *Tensor)

src/ml/tensor.go:345

↓ 6 callersFunctionAdd

(input *Tensor, other *Tensor)

src/ml/operations_impl.go:307

↓ 6 callersFunctionBFloat16bitsToFloat32

(b16 uint16)

src/dtype/bfloat16.go:55

↓ 6 callersMethodClose

()

src/common/logging.go:57

↓ 6 callersMethodConsoleFatal

(v ...any)

src/common/logging.go:41

↓ 6 callersMethodGetItem_AsFloat32

(loc []int)

src/ml/tensor.go:203

↓ 6 callersFunctionIterateOverSize

(size []int, ignoreTrailingDimensions int)

src/ml/tensoriterators.go:76

↓ 6 callersMethodReadByte

()

src/pickle/picklereader.go:58

↓ 6 callersMethodToBFloat16

()

src/ml/tensor.go:430

↓ 6 callersMethodToFloat32

(val any)

src/ml/datatypefuncset.go:15

↓ 6 callersFunctionprocessErrors

(errors ...error)

src/ml/helpers.go:8

↓ 5 callersFunctionFriendlyPanic

(err error)

src/common/utils.go:125

↓ 5 callersMethodGet

(key string)

src/pickle/types.go:22

↓ 5 callersMethodGetKeys

()

src/pickle/types.go:41

↓ 5 callersFunctionMatMul

(input *Tensor, other *Tensor)

src/ml/operations_impl.go:449

↓ 5 callersMethodSetItemByOffset

(offset int, val any)

src/ml/tensor.go:188

↓ 5 callersMethodSetItemByOffset_FromFloat32

(offset int, val float32)

src/ml/tensor.go:195

↓ 5 callersMethodSetItem_FromFloat32

(loc []int, val float32)

src/ml/tensor.go:208

↓ 5 callersMethodString

()

src/ml/tensor.go:71

↓ 5 callersMethodTokenizeString

(text string, addBeginOfSentence bool)

src/inference/tokenize.go:178

↓ 5 callersFunctionnewDataType

(name string, itemSample any, funcSet DataTypeFuncSet)

src/ml/datatype.go:26

↓ 4 callersMethodApply_AsFloat32

(fn func(val float32) float32)

src/ml/tensor.go:231

↓ 4 callersMethodBits

()

src/dtype/bfloat16.go:15

↓ 4 callersMethodGetElementCount

()

src/ml/tensor.go:59

↓ 4 callersMethodGetItemByOffset

(offset int)

src/ml/tensor.go:174

↓ 4 callersFunctionIterateOverTwo

(refTensor *Tensor, expandingTensor *Tensor, ignoreTrailingDimensions int)

src/ml/tensoriterators.go:105

↓ 4 callersFunctionPow

(input *Tensor, power float64)

src/ml/operations_impl.go:197

↓ 4 callersFunctionWaitGroupDone

(wg *sync.WaitGroup)

src/common/utils.go:19

↓ 4 callersFunctionZeros

(size []int, dataType DataType)

src/ml/operations_impl.go:66

↓ 4 callersFunctionattentionRepeatKV

(x *ml.Tensor, N_Rep int)

src/model/llamatransformer.go:529

↓ 4 callersFunctioncalculateShortenedExpectedIndex

(i int, dimensionSize int, shorten bool)

src/ml/utils.go:11

↓ 4 callersFunctioncheckSameDataType

(a *Tensor, b *Tensor)

src/ml/helpers.go:31

↓ 4 callersFunctiongetTensor

(model *Model, name string, expectedShape []int)

src/model/loader.go:183

↓ 4 callersFunctionpop

(stack []interface{})

src/pickle/pickledispatch.go:80

↓ 4 callersFunctiontestSimulatedEmojiOutput

(t *testing.T, inputStr string, expectedAssistantLines []string, expectedWaitingLines []string)

cmd/main_test.go:70

↓ 4 callersMethodupdateOutput

()

cmd/main.go:392

↓ 3 callersMethodByteFallbackString

()

src/model/model.go:98

↓ 3 callersMethodClose

()

src/torch/torchmodelreader.go:35

↓ 3 callersFunctionDuplicateTensor

(input *Tensor)

src/ml/tensor.go:53

↓ 3 callersMethodFree

()

src/model/model.go:56

↓ 3 callersMethodGetElementCount

()

src/model/model.go:60

↓ 3 callersMethodItem

()

src/ml/tensor.go:213

↓ 3 callersFunctionNewInferenceArgs

()

src/common/inferenceargs.go:7

↓ 3 callersFunctionNewLogger

(consoleWriter io.Writer, debugWriter io.Writer)

src/common/logging.go:20

↓ 3 callersFunctionNewRMSNorm

(epsilon float32, weights *ml.Tensor)

src/model/llamatransformer.go:626

↓ 3 callersFunctionReadBFloat16LittleEndian

(b []byte)

src/dtype/bfloat16.go:39

↓ 3 callersFunctioncheckIsMatrix

(t *Tensor)

src/ml/helpers.go:24

↓ 3 callersFunctioncheckIsVector

(t *Tensor)

src/ml/helpers.go:17

↓ 3 callersMethodcopyOfLoc

()

src/ml/tensoriterators.go:34

↓ 3 callersFunctionextractRuneData

(str string)

src/inference/emoji.go:42

↓ 2 callersFunctionAlmostEqualFloat32

(a float32, b float32, threshold float64)

src/common/utils.go:71

↓ 2 callersFunctionArgmax

(input *Tensor, dim int)

src/ml/operations_impl.go:513

↓ 2 callersFunctionCheckBroadcastable

(t1 *Tensor, t2 *Tensor, isCommutative bool)

src/ml/tensor.go:419

↓ 2 callersFunctionDivToScalar

(input *Tensor, scalar any)

src/ml/operations_impl.go:273

↓ 2 callersMethodFloat64

()

src/dtype/bfloat16.go:23

↓ 2 callersMethodGenerateStringGeneric

(promptTokens []model.TokenId, tokenGeneratorFn TokenGeneratorFn)

src/inference/inference.go:73

↓ 2 callersFunctionInterfaceArrToIntArr

(arr []interface{})

src/common/utils.go:59

↓ 2 callersFunctionInterfaceToInt

(val interface{})

src/common/utils.go:28

↓ 2 callersMethodIsCompatible

(val any)

src/ml/datatypefuncset.go:13

↓ 2 callersFunctionLoadModel

(modelDir string)

src/model/loader.go:18

↓ 2 callersFunctionLoadModelEx

(modelDir string, includeTensors bool, includeVocab bool)

src/model/loader.go:22

↓ 2 callersMethodLogf

(format string, v ...any)

src/model/inferencecontext.go:48

↓ 2 callersFunctionNewAppState

()

cmd/main.go:383

↓ 2 callersFunctionNewEmptyTensorLike

(input *Tensor, allocateRawData bool)

src/ml/tensor.go:49

↓ 2 callersFunctionNewInferenceContext

(model *Model, inferenceArgs common.InferenceArgs, logFn func(format string, v ...any))

src/model/inferencecontext.go:17

↓ 2 callersFunctionNewInferenceEngine

(model *model.Model, inferenceArgs common.InferenceArgs, logFn func(format string, v ...any))

src/inference/inference.go:50

↓ 2 callersFunctionNewPickleDict

()

src/pickle/types.go:16

next →1–100 of 352, ranked by callers