Functions — github.com/dromara/Omega-AI

CodeHub MCPcopy Create free account

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

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

⨍ Functions 13,024◇ Types & classes 879

↓ 36 callersMethodgetCudaManager

src/main/java/com/omega/engine/loss/LossFunction.java:56

↓ 36 callersMethodgetOutput

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

↓ 36 callersMethodgetkLinerLayer

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

↓ 36 callersMethodgetvLinerLayer

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

↓ 36 callersMethodlatend_norm

(Tensor x,Tensor mean,Tensor std)

src/main/java/com/omega/example/dit/models/IDDPM.java:759

↓ 36 callersMethodloss

(Tensor output, Tensor label)

src/main/java/com/omega/engine/nn/network/CNN.java:78

↓ 35 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sprint/FluxDiTMainMoudue_Sprint_Reg.java:433

↓ 35 callersMethodpad

src/main/java/com/omega/example/transformer/utils/tokenizers/Tokenizer.java:20

↓ 34 callersMethodback

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

↓ 34 callersMethodforward

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

↓ 34 callersMethodinitCUDAFunctions

src/main/java/com/omega/engine/gpu/CUDAModules.java:292

↓ 34 callersMethodsoftmax

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

↓ 33 callersMethodcompute_xt

@param t @param x0 is noise @param x1 is org latend alpha_t = t sigma_t = 1 - t xt = alpha_t x1 + sigma_t x0

src/main/java/com/omega/example/dit/models/ICPlan.java:1109

↓ 33 callersMethoddispatchTask

(Vector<Task<Object>> workers)

src/main/java/com/omega/common/task/TaskEngine.java:38

↓ 33 callersMethodgetHandle

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

↓ 33 callersMethodgetInstance

src/main/java/com/omega/common/task/TaskEngine.java:26

↓ 33 callersMethodmultiplication

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

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

↓ 33 callersMethodq_sample

(Tensor latend,Tensor noise,Tensor output,Tensor t)

src/main/java/com/omega/example/dit/models/IDDPM.java:753

↓ 32 callersMethodIU

src/main/java/com/omega/example/yolo/utils/YoloImageUtils.java:24

↓ 32 callersMethodaccuracyBatchFisrt

(Tensor input, Tensor output, Tensor labelData, int time, int batchSize, String[] vocab, int igonre)

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

↓ 32 callersMethodcompute_ut

@param t @param x0 is noise @param x1 is org latend @param ut is velocity d_alpha_t = 1 d_sigma_t = -1 ut = d_alpha_t x1 + d_sigma_t x0

src/main/java/com/omega/example/dit/models/ICPlan.java:1123

↓ 32 callersMethodgetName

src/main/java/com/omega/example/yolo/model/YoloImage.java:23

↓ 32 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/DiTMoudue_SRA.java:281

↓ 32 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/DiTOrgMoudue_SRA.java:282

↓ 32 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sprint/OmegaDiTMainMoudueDoubleLabel.java:422

↓ 32 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/mmjit/AttentionLayer.java:396

↓ 31 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetAttentionLayer.java:469

↓ 31 callersMethodgetqLinerLayer

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

↓ 31 callersMethodt

src/main/java/com/omega/example/dit/models/ICPlan.java:1074

↓ 31 callersMethodtensor

(int n, int c, int h, int w, float[] x, boolean isGPU)

src/main/java/com/omega/engine/tensor/Tensors.java:4

↓ 30 callersMethodback

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

↓ 30 callersMethodfreeAll

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

↓ 30 callersMethodgetDets

src/main/java/com/omega/example/yolo/model/YoloBox.java:19

↓ 30 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/org/DiTMainMoudue.java:258

↓ 30 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/block/VideoDiTConvBlockI2V.java:250

↓ 30 callersMethodgetWidth

src/main/java/com/omega/common/data/LayerConfig.java:123

↓ 30 callersMethodrandom

高斯随机数 @param x @return

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

↓ 29 callersMethodOP

(OPType opType, Tensor self, Tensor other)

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

↓ 29 callersMethodforward

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

↓ 29 callersMethodforward

src/main/java/com/omega/engine/nn/layer/diffusion/unet/UNetCond2.java:539

↓ 29 callersMethodgetHeight

src/main/java/com/omega/common/data/LayerConfig.java:115

↓ 29 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/DiTOrgMoudue2.java:314

↓ 29 callersMethodgetPointer

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

↓ 29 callersMethodone

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

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

↓ 28 callersMethodcheckCUDA

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

↓ 28 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sana/SanaDiTBlock.java:262

↓ 28 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/txt/DiT_TXTMainMoudue.java:282

↓ 28 callersMethodgetZ

src/main/java/com/omega/engine/nn/layer/jit/JiTMainMoudue_REPA.java:630

↓ 27 callersMethodcumprod

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

↓ 27 callersMethodforward

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

↓ 27 callersMethodforward

src/main/java/com/omega/engine/nn/layer/unet/UNetMidBlockLayer.java:302

↓ 27 callersMethodgetLabelSet

src/main/java/com/omega/example/yolo/utils/YoloDataLoader.java:179

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/DiTOrgMoudue.java:255

↓ 27 callersMethodgetOutput

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

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/PixArtMoudue.java:255

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sprint/FluxDiTMainMoudue_Sprint.java:404

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/block/VideoDiTBlock.java:232

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/block/VideoDiTConvBlock.java:272

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetAttentionLayer2.java:409

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetMidBlockLayer.java:248

↓ 27 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/transformer/CausalSelfAttentionLayer.java:375

↓ 27 callersMethodloadConfigToModel

(Network nn, String filepath)

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

↓ 26 callersMethodback

src/main/java/com/omega/engine/nn/layer/dit/PixArtBlock.java:502

↓ 26 callersMethodencodeInt

src/main/java/com/omega/example/transformer/utils/SentencePieceTokenizer.java:169

↓ 26 callersMethodeos

src/main/java/com/omega/example/transformer/utils/tokenizers/Tokenizer.java:18

↓ 26 callersMethodgetOutput

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

↓ 26 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/deco/FlattenDiTBlock.java:234

↓ 26 callersMethodgetUpBlocks

src/main/java/com/omega/engine/nn/network/DiffusionUNet.java:517

↓ 26 callersMethodininT

(int start,int end,int count)

src/main/java/com/omega/example/dit/models/ICPlan.java:86

↓ 26 callersMethoduniform

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

↓ 25 callersMethodaddAxisBack

(Tensor dx, Tensor dy,int N, int C, int H, int W, int axis)

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

↓ 25 callersMethodback

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

↓ 25 callersMethodback

src/main/java/com/omega/engine/nn/layer/diffusion/ResidualBlockLayer.java:330

↓ 25 callersMethodbmm

(Pointer dA, Pointer dB, Pointer dC, int batch_size, int m, int n, int k, int CUBLAS_OP_A, int CUBLAS_OP_B)

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

↓ 25 callersMethodcheckShape

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

↓ 25 callersMethodencode

src/main/java/com/omega/engine/nn/network/vae/DC_AE.java:111

↓ 25 callersMethodgetData

src/main/java/com/omega/example/yolo/data/MemeryBlock.java:29

↓ 25 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/wfvae/decoder/WFDecoder.java:410

↓ 25 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/transformer/MultiHeadAttentionLayer.java:321

↓ 25 callersMethodgetoLinerLayer

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

↓ 25 callersMethodlossDiff

(Tensor output, Tensor label)

src/main/java/com/omega/engine/nn/network/CNN.java:94

↓ 25 callersMethodshowImgs

(String outputPath, Tensor input, float[] mean, float[] std)

src/main/java/com/omega/example/dit/test/FluxDiTTest.java:2941

↓ 25 callersMethodwriteTensor

(Tensor x, FileOutputStream writer)

src/main/java/com/omega/example/dit/utils/DatasetCreater.java:69

↓ 24 callersMethodback

src/main/java/com/omega/engine/nn/layer/dit/DiTSkipBlock.java:352

↓ 24 callersMethodback

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

↓ 24 callersMethodback

src/main/java/com/omega/engine/nn/layer/dit/org/DiTBlock.java:404

↓ 24 callersMethodclamp

(int x, int min, int max)

src/main/java/com/omega/common/utils/ImageUtils.java:605

↓ 24 callersMethodforward

src/main/java/com/omega/engine/nn/layer/dit/DiTSkipBlock.java:335

↓ 24 callersMethodforward

src/main/java/com/omega/engine/nn/layer/dit/org/DiTBlock.java:387

↓ 24 callersMethodgetName

src/main/java/com/omega/common/data/NetworkConfig.java:16

↓ 24 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/DiTSkipBlock.java:248

↓ 24 callersMethodgetThreads

src/main/java/com/omega/engine/parallel/dp/DP.java:119

↓ 24 callersMethodsort

(Tensor centers)

src/main/java/com/omega/example/yolo/utils/AnchorBoxUtils.java:13

↓ 24 callersMethodsum

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

↓ 23 callersMethodback

src/main/java/com/omega/engine/nn/layer/unet/UNetAttentionLayer.java:631

↓ 23 callersMethodcheckCUDA

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

↓ 23 callersMethodclose

src/main/java/com/omega/example/transformer/utils/bpe/CNBpeTokenizer.java:199

↓ 23 callersMethodcreateLike

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

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/diffusion/unet/UNetCond2.java:505

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/sd_vae/moudles/SDVAEResidual.java:136

← previous next →201–300 of 13,024, ranked by callers