Functions — github.com/dromara/Omega-AI

hub / github.com/dromara/Omega-AI / functions

Functions13,024 in github.com/dromara/Omega-AI

⨍ Functions 13,024◇ Types & classes 879

↓ 3,703 callersMethodget

获取指定维度数据 @param position int[dims,start,count] dims: tensor 维度 0:number,1:channel,2:height,3:width

src/main/java/com/omega/engine/tensor/Tensor.java:919

↓ 3,010 callersMethodloadData

(Tensor x, Map<String, Object> weightMap, String key)

src/main/java/com/omega/example/common/ModeLoaderlUtils.java:1462

↓ 2,670 callersMethodgetGpuData

()

src/main/java/com/omega/engine/tensor/Tensor.java:551

↓ 2,042 callersMethodTensor_OP

()

src/main/java/com/omega/engine/nn/layer/Layer.java:262

↓ 1,498 callersMethodcreateGPUTensor

(Tensor t, int number, int channel, int height, int width, boolean hasGPU)

src/main/java/com/omega/engine/tensor/Tensor.java:222

↓ 981 callersMethodsize

size @param data @return

src/main/java/com/omega/common/utils/MatrixUtils.java:582

↓ 964 callersMethodadd

(Tensor x, Tensor y, Tensor output)

src/main/java/com/omega/engine/nn/layer/gpu/BasicBlockKernel.java:45

↓ 825 callersMethodsetInput

(Tensor input)

src/main/java/com/omega/engine/nn/layer/Layer.java:127

↓ 763 callersMethodsetDelta

转换并设置输入数据

src/main/java/com/omega/engine/nn/layer/Layer.java:145

↓ 734 callersMethodview

(int number, int channel, int height, int width)

src/main/java/com/omega/engine/tensor/Tensor.java:361

↓ 686 callersMethodsyncHost

()

src/main/java/com/omega/engine/tensor/Tensor.java:559

↓ 639 callersMethodpermute

(Tensor input, Tensor query, Tensor key, Tensor value, int B, int N, int NH, int d)

src/main/java/com/omega/engine/nn/layer/gpu/AttentionKernel.java:301

↓ 616 callersMethodviewOrg

(int number, int channel, int height, int width)

src/main/java/com/omega/engine/tensor/Tensor.java:369

↓ 594 callersMethodaddLayer

(Layer layer)

src/main/java/com/omega/engine/nn/network/Network.java:203

↓ 594 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/network/Network.java:159

↓ 577 callersMethodhostToDevice

()

src/main/java/com/omega/engine/tensor/Tensor.java:571

↓ 544 callersMethodgetLocalFunctionByModule

(String fileName, String functionName)

src/main/java/com/omega/engine/gpu/CUDAManager.java:125

↓ 505 callersMethodgetCudaManager

()

src/main/java/com/omega/engine/gpu/BaseKernel.java:526

↓ 467 callersMethodloadLabel_offset

(Tensor label, int index, String labelStr)

src/main/java/com/omega/example/sd/utils/SDImageDataLoaderEN.java:371

↓ 463 callersMethodmul

(Tensor y)

src/main/java/com/omega/engine/tensor/Tensor.java:818

↓ 433 callersMethodgradientCheck

()

src/main/java/com/omega/engine/nn/layer/normalization/gpu/BNKernel.java:221

↓ 421 callersMethodclearGPU

(cudaStream_t stream)

src/main/java/com/omega/engine/tensor/Tensor.java:675

↓ 362 callersMethodshape

()

src/main/java/com/omega/engine/tensor/Tensor.java:398

↓ 361 callersMethodcreate

create instance @param type @return none null momentum adam

src/main/java/com/omega/engine/updater/UpdaterFactory.java:27

↓ 361 callersMethodsetUpdater

(Updater updater)

src/main/java/com/omega/engine/nn/layer/Layer.java:106

↓ 346 callersMethodgetLastLayer

()

src/main/java/com/omega/engine/nn/network/Network.java:169

↓ 341 callersMethodgetByChannel

(Tensor org, Tensor target, int start, int count, Tape tape)

src/main/java/com/omega/engine/ad/op/data/GetOP.java:97

↓ 323 callersMethodsum

@param x @return @throws @Title: count @Description: TODO(这里用一句话描述这个方法的作用)

src/main/java/com/omega/common/utils/MatrixOperation.java:1486

↓ 318 callersMethodadd

(Tape tape)

src/main/java/com/omega/engine/ad/Graph.java:695

↓ 308 callersMethodgetDataLength

()

src/main/java/com/omega/engine/tensor/Tensor.java:442

↓ 303 callersMethodgetCache

(String key, int[] shape)

src/main/java/com/omega/engine/gpu/CUDAMemoryManager.java:34

↓ 288 callersMethodgetLayerType

()

src/main/java/com/omega/engine/nn/layer/Layer.java:100

↓ 281 callersMethodreadJsonFileSmallWeight

(String path)

src/main/java/com/omega/example/transformer/utils/LagJsonReader.java:53

↓ 274 callersMethodpow

()

src/main/java/com/omega/engine/tensor/Tensor.java:858

↓ 271 callersMethodcreateTensor

(Tensor t, int number, int channel, int height, int width, float[] data, boolean hasGPU)

src/main/java/com/omega/engine/tensor/Tensor.java:202

↓ 259 callersMethodisHasGPU

()

src/main/java/com/omega/engine/tensor/Tensor.java:700

↓ 240 callersMethodshowDM

()

src/main/java/com/omega/engine/tensor/Tensor.java:598

↓ 228 callersMethodcuda

()

src/main/java/com/omega/engine/nn/layer/Layer.java:266

↓ 228 callersMethoddiff

()

src/main/java/com/omega/engine/nn/layer/Layer.java:82

↓ 211 callersMethodshowDM

(Pointer d, float[] data)

src/main/java/com/omega/engine/nn/layer/gpu/ConvKernel.java:345

↓ 204 callersMethodgetOnceSize

()

src/main/java/com/omega/engine/tensor/Tensor.java:450

↓ 190 callersMethodbmmEX

(int transa, int transb, int m, int n, int k, float alpha, /** host or device pointer */

src/main/java/com/omega/engine/gpu/GPUOP.java:492

↓ 187 callersMethodtoJson

(Object object)

src/main/java/com/omega/common/utils/JsonUtils.java:8

↓ 184 callersMethodloadModel

(MMJiT model, String inputPath)

src/main/java/com/omega/example/transformer/utils/ModelUtils.java:78

↓ 181 callersMethodsqrt

(Tensor a, Tensor c)

src/main/java/com/omega/engine/ad/op/TensorOP.java:410

↓ 179 callersMethodloadWeight

(Map<String, Object> weightMap, WFVAE network, boolean showLayers)

src/main/java/com/omega/example/common/ModeLoaderlUtils.java:32

↓ 177 callersMethodgetInstance

()

src/main/java/com/omega/engine/gpu/GPUOP.java:61

↓ 177 callersMethodsaveModel

(MMJiT model, String outpath)

src/main/java/com/omega/example/transformer/utils/ModelUtils.java:58

↓ 169 callersMethodGPU_OP

()

src/main/java/com/omega/engine/nn/layer/Layer.java:258

↓ 158 callersMethodclampSelf

(float[] x, float min, float max)

src/main/java/com/omega/common/utils/MatrixOperation.java:1878

↓ 158 callersMethodloadData

(String filepath)

src/main/java/com/omega/engine/model/ModelLoader.java:369

↓ 157 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTBlock.java:393

↓ 156 callersMethodgetByNumber

(int n)

src/main/java/com/omega/engine/tensor/Tensor.java:485

↓ 153 callersMethodgaussianRandom

高斯随机数 @param x @return

src/main/java/com/omega/common/utils/RandomUtils.java:331

↓ 141 callersMethodorder

xavier随机数 @param x @return

src/main/java/com/omega/common/utils/RandomUtils.java:847

↓ 134 callersMethodcopyGPU

()

src/main/java/com/omega/engine/tensor/Tensor.java:272

↓ 131 callersMethodsetInputData

(Tensor input)

src/main/java/com/omega/engine/nn/network/Network.java:149

↓ 131 callersMethodtoString

()

src/main/java/com/omega/example/yolo/test/deepsort/Rect.java:16

↓ 129 callersMethodforward

(Tape tape)

src/main/java/com/omega/engine/ad/op/OP.java:15

↓ 128 callersMethodisWarmUp

()

src/main/java/com/omega/engine/optimizer/Optimizer.java:1756

↓ 127 callersMethodabs

(Tensor x,Tensor y)

src/main/java/com/omega/engine/ad/op/TensorOP.java:930

↓ 121 callersMethodsetByChannel

(Tensor org, Tensor target,int[] shape, int start)

src/main/java/com/omega/engine/ad/op/TensorOP.java:872

↓ 118 callersMethodshowShape

()

src/main/java/com/omega/engine/tensor/Tensor.java:402

↓ 116 callersMethodgetCosAndSin2D

(int time, int dim, int headNum)

src/main/java/com/omega/engine/nn/layer/gpu/RoPEKernel.java:281

↓ 112 callersMethodorder

@param size @return @throws @Title: zero @Description: TODO(这里用一句话描述这个方法的作用)

src/main/java/com/omega/common/utils/MatrixUtils.java:113

↓ 109 callersMethodshuffle

()

src/main/java/com/omega/example/sd/utils/SDImageLoader.java:105

↓ 107 callersMethodcheckCUDA

(int code)

src/main/java/com/omega/engine/ad/op/gpu/OPKernel.java:2309

↓ 103 callersMethodCAFFE_GET_BLOCKS

(int N)

src/main/java/com/omega/engine/ad/op/gpu/OPKernel.java:2301

↓ 100 callersMethodsetLossDiff

(Tensor lossDiff)

src/main/java/com/omega/engine/nn/network/Network.java:154

↓ 99 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/modules/DiTAttentionLayer2.java:637

↓ 97 callersMethodgetByChannel_back

(Tensor org, Tensor target, int start, int count)

src/main/java/com/omega/engine/ad/op/TensorOP.java:893

↓ 96 callersMethodval

@param size @return @throws @Title: zero @Description: TODO(这里用一句话描述这个方法的作用)

src/main/java/com/omega/common/utils/MatrixUtils.java:248

↓ 92 callersMethodcreate

create instance @param type @return none null momentum adam

src/main/java/com/omega/engine/loss/LossFactory.java:29

↓ 89 callersMethodupdateLR

(int[] lr_step)

src/main/java/com/omega/engine/optimizer/Optimizer.java:197

↓ 87 callersMethodgetGrad

()

src/main/java/com/omega/engine/tensor/Tensor.java:719

↓ 86 callersMethodreadJsonFileBigWeightIterator

(String path)

src/main/java/com/omega/example/transformer/utils/LagJsonReader.java:67

↓ 86 callersMethodsqrt

sqrt @return

src/main/java/com/omega/common/utils/MatrixOperation.java:2506

↓ 85 callersMethodgetEncoders

()

src/main/java/com/omega/engine/nn/layer/clip/CLIPTextTransformer.java:267

↓ 84 callersMethodcat_batch

(Tensor a, Tensor b, Tensor c)

src/main/java/com/omega/engine/ad/op/TensorOP.java:661

↓ 83 callersMethodfree

()

src/main/java/com/omega/engine/gpu/CUDAMemoryManager.java:149

↓ 82 callersMethodzero

@param size @return @throws @Title: zero @Description: TODO(这里用一句话描述这个方法的作用)

src/main/java/com/omega/common/utils/MatrixUtils.java:51

↓ 81 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/modules/DiTFlashAttentionLayer.java:527

↓ 81 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/sprint/MMDiTAttention.java:627

↓ 79 callersMethodsetData

(float[] data)

src/main/java/com/omega/engine/tensor/Tensor.java:458

↓ 79 callersMethodwrite

(List<String> txtList, String[] ids, FileWriter writer, Tokenizer tokenizer)

src/main/java/com/omega/example/transformer/utils/tokenizers/BatchTokenizerUtils.java:356

↓ 78 callersMethodsub

(Tensor y)

src/main/java/com/omega/engine/tensor/Tensor.java:806

↓ 77 callersMethodclear

()

src/main/java/com/omega/example/yolo/utils/OMImage.java:29

↓ 77 callersMethodclear

()

src/main/java/com/omega/engine/gpu/GPUOP.java:198

↓ 76 callersMethodgetLossType

()

src/main/java/com/omega/engine/nn/model/NetworkInit.java:103

↓ 76 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/DiTOrgBlock.java:347

↓ 76 callersMethodgetTensorOP

()

src/main/java/com/omega/engine/ad/Tape.java:179

↓ 76 callersMethodsetChannel

(int channel)

src/main/java/com/omega/engine/nn/network/Network.java:292

↓ 76 callersMethodsetWidth

(int width)

src/main/java/com/omega/engine/nn/network/Network.java:308

↓ 75 callersMethodcudaRandn

(Tensor x)

src/main/java/com/omega/engine/gpu/GPUOP.java:694

↓ 75 callersMethodsetHeight

(int height)

src/main/java/com/omega/engine/nn/network/Network.java:300

↓ 73 callersMethodgetCosAndSin

(int time, int dim, int headNum)

src/main/java/com/omega/engine/nn/layer/gpu/RoPEKernel.java:231

↓ 73 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/sprint/DiTBlock.java:308

↓ 73 callersMethodupdate

(Layer layer)

src/main/java/com/omega/engine/updater/Adam.java:20

↓ 72 callersMethodback

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTBlock.java:709

↓ 72 callersMethodget_full_clip_prompt_embeds

(Tensor input)

src/main/java/com/omega/engine/nn/network/ClipTextModel.java:122

next →1–100 of 13,024, ranked by callers