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

↓ 72 callersMethodmul_left_back

(Tensor b, Tensor delta, Tensor da, int N, int C, int H, int W, int axis)

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

↓ 72 callersMethodmul_right_back

(Tensor a, Tensor delta, Tensor db, int N, int C, int H, int W, int axis)

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

↓ 71 callersMethodgetInt

(String val)

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

↓ 71 callersMethodgetOutput

()

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

↓ 71 callersMethodupdateLRDynamic

(int it, int count, float min)

src/main/java/com/omega/engine/optimizer/MBSGDOptimizer.java:16131

↓ 70 callersMethodgetHandle

()

src/main/java/com/omega/engine/gpu/cudnn/CudnnHandleManager.java:10

↓ 70 callersMethodsubtraction

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

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

↓ 70 callersMethodxavier_uniform

(Tensor data,float gain,int fan_in, int fan_out)

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

↓ 69 callersMethodforward

()

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

↓ 68 callersMethodloadLabel_offset

(SentencePieceTokenizer tokenizer, Tensor label, Tensor mask, int index, String labelStr, int maxContextLen)

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

↓ 67 callersMethodlatend_un_norm

(Tensor x,Tensor mean,Tensor std)

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

↓ 66 callersMethodaddAxis

(Tensor a, Tensor b, Tensor c, int axis)

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

↓ 66 callersMethodgetOutput

()

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

↓ 65 callersMethodcopy_gpu

(Pointer a, Pointer b, int N, int incx, int incy)

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

↓ 65 callersMethodgetqLinerLayer

()

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

↓ 64 callersMethodunpermute

(Tensor input, Tensor output, int B, int N, int NH, int d)

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

↓ 62 callersMethodgetHeight

()

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

↓ 62 callersMethodgetWidth

()

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

↓ 62 callersMethodshowImgs

(String outputPath, Tensor input)

src/main/java/com/omega/engine/optimizer/MBSGDOptimizer.java:250

↓ 61 callersMethodback

(Tensor lossDiff)

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

↓ 59 callersMethoddecode

(Tensor latent)

src/main/java/com/omega/engine/nn/network/vae/VA_VAE.java:144

↓ 59 callersMethodgetNetwork

()

src/main/java/com/omega/engine/parallel/dp/NetworkRunnable.java:661

↓ 59 callersMethodisRequiresGrad

()

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

↓ 58 callersMethodencodeInt

(String txt)

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

↓ 58 callersMethodgetWidth

()

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

↓ 56 callersMethodload

(String path, String extName, String[] names, int[] indexs, int batchSize, Tensor input, boolean gray, boolean

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

↓ 56 callersMethodsetCurrentLabel

(Tensor currentLabel)

src/main/java/com/omega/engine/nn/layer/SoftmaxWithCrossEntropyLayer.java:115

↓ 55 callersMethodgradientCheck

@param x @return @Title: gradientCheck @Description: TODO(这里用一句话描述这个方法的作用) <p> gradientCheck: <p> (f(x + eta) - f(x - eta)) / (2 eta) ≈ f'(x)

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

↓ 54 callersMethoddecode

(Tensor latent)

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

↓ 54 callersMethodgetChannel

()

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

↓ 54 callersMethodloadLabel_offset

(BPETokenizerEN tokenizer, Tensor label, int index, int maxContextLen, String labelStr)

src/main/java/com/omega/example/dit/test/MMJiTTest.java:224

↓ 53 callersMethodconcat_channel_forward

(Tensor x1, Tensor x2, Tensor output, int B, int C1, int C2, int H, int W)

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

↓ 53 callersMethodgetDevice

(int size)

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

↓ 53 callersMethodgetHeight

()

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

↓ 53 callersMethodgetOutput

()

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

↓ 53 callersMethodgetqLinerLayer

()

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

↓ 53 callersMethodprintImage

print matrix @param data

src/main/java/com/omega/common/utils/PrintUtils.java:16

↓ 52 callersMethodsetRequiresGrad

(boolean requiresGrad)

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

↓ 51 callersMethodconcat_channel_backward

(Tensor diff, Tensor dx1, Tensor dx2, int B, int C1, int C2, int H, int W)

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

↓ 51 callersMethodgetInstance

()

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

↓ 51 callersMethodrun

(RecursiveAction action)

src/main/java/com/omega/common/task/ForkJobEngine.java:21

↓ 50 callersMethodcopy

()

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

↓ 50 callersMethodmin

(float[] x)

src/main/java/com/omega/common/utils/MathUtils.java:220

↓ 49 callersMethodcos

(float[] x)

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

↓ 49 callersMethodgetOrgShape

()

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

↓ 49 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/sprint/OmegaDiTMainMoudue_Sprint.java:510

↓ 49 callersMethodrandomInts

生成随机数 @param length @return

src/main/java/com/omega/common/utils/MathUtils.java:48

↓ 48 callersMethodforward

()

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

↓ 48 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTBlock2.java:268

↓ 48 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/sprint/FluxDiTMainMoudue_Sprint4.java:507

↓ 48 callersMethodgetX

()

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

↓ 48 callersMethodgetY_embedding

()

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

↓ 48 callersMethodsin

(float[] x)

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

↓ 47 callersMethodgetOutput

()

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

↓ 47 callersMethodloadParams

(RandomAccessFile inputStream, Tensor data)

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

↓ 47 callersMethodmaxIndex

@return @throws @Title: maxIndex @Description:

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

↓ 46 callersMethodloadData

(Tensor x, Object meta, String key)

src/main/java/com/omega/example/diffusion/test/DiffusionModelTest.java:168

↓ 46 callersMethodsaveParams

(RandomAccessFile outputStream, Tensor data)

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

↓ 45 callersMethodforward_with_cfg

(MMDiT_RoPE dit, Tensor y0, Tensor t, Tensor context, Tensor cos, Tensor sin, Tensor y1, Tensor eps, float cfg

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

↓ 44 callersMethodcreateRGBImage

通过BufferedImage图片流调整图片大小 <p> 指定压缩后长宽

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

↓ 44 callersMethodexp

()

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

↓ 44 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTMainMoudue_REPA.java:329

↓ 44 callersMethodgetkLinerLayer

()

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

↓ 44 callersMethodgetvLinerLayer

()

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

↓ 44 callersMethodlog

()

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

↓ 44 callersMethodmultiplyFloat

Multiplies the matrices A and B and writes the result into C. @param size The size of the matrices @param A Matrix A @param B Matrix B @param C

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

↓ 43 callersMethodforward

()

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

↓ 43 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/sprint/OmegaDiTMainMoudue_Sprint2.java:482

↓ 43 callersMethodunpermute_backward

(Tensor din, Tensor dout, int B, int N, int NH, int d)

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

↓ 42 callersMethodbytesToHex

change bytes into a hex string. @param bytes bytes @return the returned hex string

src/main/java/com/omega/common/utils/DataLoader.java:889

↓ 42 callersMethodcat_bacth_copy

(Tensor c, Tensor a, Tensor b)

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

↓ 42 callersMethodforward_with_path_drop_cfg

(FluxDiT_SPRINT dit, Tensor y0, Tensor t, Tensor context, Tensor null_context, Tensor cos, Tensor sin, Tensor

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

↓ 42 callersMethodgetOutput

()

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

↓ 42 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTMainMoudue_REG.java:308

↓ 42 callersMethodgetOutput

()

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

↓ 41 callersMethodgetBbox

()

src/main/java/com/omega/example/yolo/model/YoloDetection.java:25

↓ 40 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/flux/FluxDiTMainMoudue.java:313

↓ 40 callersMethodgetOutput

()

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

↓ 40 callersMethodgetSize

()

src/main/java/com/omega/engine/gpu/GPUWorkspace.java:17

↓ 40 callersMethodloadLabel_offset

(BPETokenizerEN tokenizer, Tensor label, int index, String labelStr)

src/main/java/com/omega/example/dit/dataset/ImageClipDataLoader.java:324

↓ 40 callersMethodtransform

transform @param x c h w @return @index ci h w + hi w + wi

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

↓ 39 callersMethodadd

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

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

↓ 39 callersMethodgetChannel

()

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

↓ 39 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/jit/JiTMainMoudue.java:310

↓ 39 callersMethodgetOutput

()

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

↓ 39 callersMethodgetPointer

()

src/main/java/com/omega/engine/gpu/GPUWorkspace.java:9

↓ 39 callersMethodmax

(int a, int b)

src/main/java/com/omega/engine/gpu/JNITest.java:18

↓ 38 callersMethodappend

(Tensor input, Tensor output, int length, int count, int currentIndex)

src/main/java/com/omega/engine/nn/layer/opensora/wfvae/modules/WFKernel.java:172

↓ 38 callersMethodclose

关闭录制器并完成视频写入

src/main/java/com/omega/example/opensora/utils/VideoWriter.java:244

↓ 38 callersMethodforward

()

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

↓ 38 callersMethodgetEncoder

()

src/main/java/com/omega/engine/nn/network/ClipVision.java:166

↓ 38 callersMethodgetOutput

()

src/main/java/com/omega/engine/nn/layer/dit/mmdit/DiTJoinBlockHead.java:431

↓ 38 callersMethodlinspace

(float start, float end, int count)

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

↓ 37 callersMethodaxpy_gpu

(Tensor a, Tensor b, int N, float ALPHA, int INCX, int INCY)

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

↓ 37 callersMethodclose

()

src/main/java/com/omega/example/transformer/dataset/SFTDataset2.java:167

↓ 37 callersMethodcolor2rgb2

(float[] data, int height, int width)

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

↓ 37 callersMethoddiv

(Tensor y)

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

↓ 37 callersMethodgather

(float[] v, int[] index)

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

↓ 37 callersMethodgetAttn

()

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

↓ 36 callersMethodforward2d

(Tensor cos, Tensor sin, Tensor input, Tensor output,int T,int HN,int HS)

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

← previous next →101–200 of 13,024, ranked by callers