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

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/dit/moudles/OpenSoraDiTBlock.java:231

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/vqvae/tiny/VQVAEResidual.java:136

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sprint/FluxDiTMainMoudue_Sprint3.java:400

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/OmegaVideoDiTMain2.java:356

↓ 23 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/OmegaVideoDiTMain.java:358

↓ 23 callersMethodgetkLinerLayer

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

↓ 23 callersMethodloadImgDataToArray

(String filePath)

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

↓ 23 callersMethodpermute

(Tensor a, Tensor b, int[] permutes)

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

↓ 22 callersMethodCAFFE_GET_BLOCKS

src/main/java/com/omega/engine/nn/network/vae/VAEKernel.java:138

↓ 22 callersMethodback

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

↓ 22 callersMethodclearGrad

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

↓ 22 callersMethodforward

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

↓ 22 callersMethodforward

src/main/java/com/omega/engine/nn/layer/unet/UNetUpBlockLayer.java:350

↓ 22 callersMethodgetMemoryManager

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

↓ 22 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/diffusion/unet/UNetAttentionBlock.java:274

↓ 22 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/vqvae/tiny/VQVAEAttentionLayer.java:343

↓ 22 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetUpBlockLayer.java:296

↓ 22 callersMethodgetY

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

↓ 22 callersMethodgetoLinerLayer

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

↓ 22 callersMethodgetqLinerLayer

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

↓ 22 callersMethodsetOpType

(OPType opType)

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

↓ 22 callersMethodshowImgs

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

src/main/java/com/omega/example/dit/test/OmegaDiTTest.java:1518

↓ 22 callersMethodshowImgs

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

src/main/java/com/omega/example/dit/test/DiTTest.java:1559

↓ 22 callersMethodsoftmax_backward

N = B NH @param input @param output @param scale @param B @param T

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

↓ 22 callersMethoduniformFloat

(float min, float max)

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

↓ 21 callersMethodCAFFE_GET_BLOCKS

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

↓ 21 callersMethodback

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

↓ 21 callersMethodclose

src/main/java/com/omega/example/transformer/utils/CNWikiTokenizer4.java:175

↓ 21 callersMethodcompute_v

(Tensor x, Tensor t, Tensor z, Tensor v, float t_eps)

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

↓ 21 callersMethodcos

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

↓ 21 callersMethodforward

src/main/java/com/omega/engine/nn/network/Llama2.java:107

↓ 21 callersMethodforward

src/main/java/com/omega/engine/nn/layer/diffusion/unet/UNetAttentionBlock.java:320

↓ 21 callersMethodgetClass

src/main/java/com/omega/example/yolo/test/YoloV1Test.java:122

↓ 21 callersMethodgetImageData

读取一张图片的RGB值 @throws Exception

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

↓ 21 callersMethodgetMlp

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

↓ 21 callersMethodgetNorm1

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

↓ 21 callersMethodgetNorm2

src/main/java/com/omega/engine/nn/layer/clip/CLIPEncoderLayer.java:274

↓ 21 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/vae/modules/AttentionBlock3D.java:298

↓ 21 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/dit/moudles/OpenSoraDiTMoudue.java:263

↓ 21 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDDTMainMoudue.java:276

↓ 21 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sana/GLUMBConv.java:136

↓ 21 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sana/LiteLALayer.java:256

↓ 21 callersMethodget_clip_prompt_embeds

(Tensor input, Tensor eos_idx, Tensor embeds)

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

↓ 21 callersMethodgetqLinerLayer

src/main/java/com/omega/engine/nn/layer/llama/LlamaCausalSelfAttention2Layer.java:562

↓ 21 callersMethodgetvLinerLayer

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

↓ 21 callersMethodloadImage

(String filePath, float[] data)

src/main/java/com/omega/example/yolo/data/ImageLoader.java:833

↓ 21 callersMethodmulPlus

(Tensor a, Tensor b, Tensor c)

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

↓ 20 callersMethodCAFFE_GET_BLOCKS

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

↓ 20 callersMethodbackward

(Tensor cos, Tensor sin, Tensor delta, Tensor diff)

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

↓ 20 callersMethodforward

src/main/java/com/omega/engine/nn/layer/dit/deco/MMDiTAttentionLayer.java:577

↓ 20 callersMethodgetDownBlocks

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

↓ 20 callersMethodgetInput

src/main/java/com/omega/example/unet/utils/SegImageLoader2.java:219

↓ 20 callersMethodgetInput

src/main/java/com/omega/example/unet/utils/SegImageLoader.java:198

↓ 20 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/asr/MultiHeadAttentionMaskLayer.java:329

↓ 20 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/diffusion/unet/UNetCrossAttentionLayer.java:335

↓ 20 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/sprint/FluxDiTMainMoudue_Tread.java:383

↓ 20 callersMethodgetOutput

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

↓ 20 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/block/VideoDiTAttentionLayer.java:420

↓ 20 callersMethodlinear_interp

(TensorOP tensorOP,float t0, float t1, Tensor y0, Tensor y1, float t)

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

↓ 20 callersMethoduniformInt

(int max,Tensor data)

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

↓ 19 callersMethodcat_width

(Tensor a, Tensor b, Tensor c)

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

↓ 19 callersMethodcheckCUDA

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

↓ 19 callersMethodencodeInt

src/main/java/com/omega/example/transformer/utils/bpe/BPETokenizer3.java:260

↓ 19 callersMethodforward

src/main/java/com/omega/engine/nn/layer/dit/modules/DiTCrossAttentionLayer2.java:544

↓ 19 callersMethodforward

src/main/java/com/omega/engine/nn/layer/unet/UNetDownBlockLayer.java:319

↓ 19 callersMethodgetNumber

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

↓ 19 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/LSTMLayer.java:242

↓ 19 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/wfvae/encoder/WFEncoder.java:300

↓ 19 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/mmdit/MMDiTMoudue.java:243

↓ 19 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetCond.java:543

↓ 19 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/unet/UNetDownBlockLayer.java:270

↓ 19 callersMethodgetqLinerLayer

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

↓ 19 callersMethodgetqLinerLayer

src/main/java/com/omega/engine/nn/layer/t5/T5AttentionLayer.java:374

↓ 19 callersMethodhandle

Handle. @param returnCode the return run

src/main/java/com/omega/engine/gpu/cudnn/TestJCudnn.java:454

↓ 19 callersMethodloadData

src/main/java/com/omega/example/dit/dataset/LatendDataset.java:109

↓ 18 callersMethodback

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

↓ 18 callersMethodback

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

↓ 18 callersMethodback

src/main/java/com/omega/engine/nn/layer/dit/txt/DiT_TXTBlock.java:442

↓ 18 callersMethodforward

src/main/java/com/omega/engine/nn/network/Llama3.java:134

↓ 18 callersMethodforward

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

↓ 18 callersMethodforward

src/main/java/com/omega/engine/nn/layer/dit/txt/DiT_TXTBlock.java:425

↓ 18 callersMethodforward

src/main/java/com/omega/engine/nn/layer/unet/UNetCond.java:608

↓ 18 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/llama/LlamaCausalSelfAttentionLayer.java:384

↓ 18 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/llama/LlamaCausalSelfAttention2Layer.java:377

↓ 18 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/clip/CLIPAttentionLayer.java:367

↓ 18 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/opensora/wfvae/modules/WFAttentionBlock3D.java:306

↓ 18 callersMethodgetOutput

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

↓ 18 callersMethodgetOutput

src/main/java/com/omega/engine/nn/layer/dit/video/block/GLUMBConv.java:211

↓ 18 callersMethodgetPatchEmbedding

src/main/java/com/omega/engine/nn/layer/dit/DiTPatchEmbeddingLayer.java:235

↓ 18 callersMethodgetPositions

(int b, int time)

src/main/java/com/omega/example/transformer/utils/CNChatTokenizer.java:71

↓ 18 callersMethodgetWidth

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

↓ 18 callersMethodgetkLinerLayer

src/main/java/com/omega/engine/nn/layer/t5/T5AttentionLayer.java:382

↓ 18 callersMethodgetqLinerLayer

src/main/java/com/omega/engine/nn/layer/llama/LlamaCausalSelfAttentionLayer.java:585

↓ 18 callersMethodgetqLinerLayer

src/main/java/com/omega/engine/nn/layer/dit/deco/MMDiTAttentionLayer.java:750

↓ 18 callersMethodgetqLinerLayer

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

↓ 18 callersMethodgetvLinerLayer

src/main/java/com/omega/engine/nn/layer/t5/T5AttentionLayer.java:390

↓ 18 callersMethodrandomFloat

(float min, float max)

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

↓ 18 callersMethodreadFloatArray

(RandomAccessFile inputStream, float[] data)

src/main/java/com/omega/engine/nn/network/utils/ModelUtils.java:95

↓ 18 callersMethodroll

(Tensor x, Tensor out, int shifts, int dims)

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

↓ 18 callersMethodsetColor

src/main/java/com/omega/engine/nn/data/ImageData.java:142

← previous next →301–400 of 13,024, ranked by callers