Functions — github.com/InfiniTensor/InfiniTensor

CodeHub MCPcopy Create free account

hub / github.com/InfiniTensor/InfiniTensor / functions

Functions2,429 in github.com/InfiniTensor/InfiniTensor

⨍ Functions 2,429◇ Types & classes 632

↓ 1 callersFunctionshard_gemm

(node: NodeProto)

examples/distributed/parallel.py:43

↓ 1 callersFunctionshard_node

(node: NodeProto)

examples/distributed/parallel_opt.py:138

↓ 1 callersFunctionshard_reshape

(node: NodeProto)

examples/distributed/parallel_opt.py:81

↓ 1 callersFunctionshard_split

(node: NodeProto)

examples/distributed/parallel_opt.py:121

↓ 1 callersFunctionshard_transpose

(node: NodeProto)

examples/distributed/parallel_opt.py:132

↓ 1 callersMethodsimplify

src/nnet/Visitor/SimplifyFormulaMutator.cc:22

↓ 1 callersMethodslice

src/core/graph_handler.cc:482

↓ 1 callersMethodsqueeze

src/core/graph_handler.cc:674

↓ 1 callersFunctionstart_single

(name, model, data_type)

examples/distributed/bang/bang_launch.py:110

↓ 1 callersFunctiontestConvCudnnFP16

test/kernels/cuda/test_cuda_conv_fp16.cc:13

↓ 1 callersFunctiontestConvTransposedAclnn

test/kernels/ascend/test_ascend_conv_transpose_2d.cc:11

↓ 1 callersFunctiontestConvTransposedCudnn

test/kernels/cuda/test_cuda_conv_transposed_2d.cc:13

↓ 1 callersFunctiontestConvTransposedMkl

test/kernels/intelcpu/test_mkl_conv_transposed.cc:11

↓ 1 callersFunctiontestConvTransposedNHWCCudnn

test/kernels/cuda/test_cuda_conv_transposed_2d.cc:49

↓ 1 callersFunctiontestConvTransposedNHWCXdnn

test/kernels/kunlun/test_kunlun_conv_trans.cc:49

↓ 1 callersFunctiontestConvTransposedXdnn

test/kernels/kunlun/test_kunlun_conv_trans.cc:13

↓ 1 callersFunctiontestFlatten

test/kernels/ascend/test_ascend_reshape.cc:43

↓ 1 callersFunctiontestLeakyRelu

test/kernels/ascend/test_ascend_unary.cc:79

↓ 1 callersFunctiontestNet

test/kernels/bang/test_bang_net.cc:12

↓ 1 callersFunctiontest_instancenormFp32

test/kernels/ascend/test_ascend_instancenorm.cc:11

↓ 1 callersFunctiontest_layernormFp16

test/kernels/cuda/test_cuda_layernorm.cc:80

↓ 1 callersMethodtoOutputShape

src/nnet/expr.cc:43

↓ 1 callersMethodtoReadable

src/nnet/derivator.cc:318

↓ 1 callersMethodtoString

src/core/runtime.cc:163

↓ 1 callersMethodto_json

src/kernels/cuda/conv.cc:17

↓ 1 callersMethodto_onnx

(self, name: str)

pyinfinitensor/src/pyinfinitensor/onnx.py:1137

↓ 1 callersMethodtranspose

src/nnet/Visitor/MatmulTransposeMutator.cc:5

↓ 1 callersMethodtune

src/kernels/cuda/conv.cc:176

↓ 1 callersMethodunsqueeze

src/core/graph_handler.cc:685

MethodASCENDRuntimeObj

include/ascend/ascend_runtime.h:16

MethodActivationBackwardObj

src/operators/activation_backward.cc:4

MethodAllGatherObj

src/operators/all_gather.cc:4

MethodAllReduceAvgObj

src/operators/all_reduce.cc:43

MethodAllReduceBaseObj

src/operators/all_reduce.cc:4

MethodAllReduceMaxObj

src/operators/all_reduce.cc:40

MethodAllReduceMinObj

src/operators/all_reduce.cc:37

MethodAllReduceProdObj

src/operators/all_reduce.cc:34

MethodAllReduceSumObj

src/operators/all_reduce.cc:31

MethodAttentionKVCacheObj

src/operators/attention_kvcache.cc:5

MethodAvgPoolObj

include/operators/pooling.h:76

MethodBangRuntimeObj

include/bang/bang_runtime.h:17

MethodBatchNormObj

src/operators/batch_norm.cc:4

MethodBinaryOpNode

include/nnet/expr.h:292

include/core/blob.h:16

MethodBroadcastObj

src/operators/broadcast.cc:4

FunctionCHECK_CUDA_KERNEL_ERROR

src/cuda/cuda_runtime.cc:11

MethodCUDAStream

include/cuda/cuda_common.h:117

src/operators/unary.cc:149

MethodCheckIndexValid

TODO:should check everytime index updated.

src/operators/gather.cc:33

MethodCheckOOBVisitor

include/nnet/Visitor/CheckOOBVisitor.h:11

src/operators/unary.cc:33

MethodCloneMutator

include/nnet/Visitor/CloneMutator.h:9

MethodCnclCommunicatorObj

include/bang/cncl_communicator.h:20

MethodCommunicatorObj

include/core/communicator.h:14

MethodCompareMultiFormulasVisitor

include/nnet/Visitor/CompareMultiFormulasVisitor.h:11

MethodConcatObj

src/operators/concat.cc:5

MethodConstantNode

include/nnet/expr.h:317

MethodConv3dObj

src/operators/conv.cc:142

MethodConvBackwardFilterObj

src/operators/conv.cc:302

MethodConvBaseObj

src/operators/conv.cc:5

include/nnet/routine.h:69

src/operators/conv.cc:63

MethodConvTransposed2dNHWCObj

src/operators/conv.cc:361

MethodConvTransposed2dObj

src/operators/conv.cc:223

MethodCountRoutineVisitor

include/nnet/Visitor/CountRoutineVisitor.h:11

MethodCpuRuntimeObj

include/core/runtime.h:103

MethodCudaRuntimeObj

include/cuda/cuda_runtime.h:22

include/nnet/dlt.h:20

MethodDLTReorder

include/nnet/dlt.h:24

include/nnet/dlt.h:16

FIXME: default ctor should be deleted but json requires it. Solution: https://github.com/nlohmann/json#how-can-i-use-get-for-non-default-constructible

include/core/data_type.h:59

MethodDebugOutput

include/nnet/dbg.h:701

MethodDepthToSpaceObj

src/operators/transpose.cc:55

MethodDerivator

src/nnet/derivator.cc:138

src/operators/det.cc:4

MethodDropoutObj

src/operators/dropout.cc:5

MethodDummyMutator

include/core/dummy_mutator.h:8

MethodElementWiseCudnn

src/kernels/cuda/element_wise.cc:90

MethodElementWiseNode

_outputShape is redundent, but expr is still missing for DLT.

include/nnet/routine.h:86

MethodElementWiseObj

src/operators/element_wise.cc:5

src/operators/unary.cc:346

MethodException

src/utils/exception.cc:12

MethodExpandObj

src/operators/expand.cc:6

MethodExprTreeVisitor

include/nnet/visitor.h:115

MethodExtendObj

src/operators/extend.cc:6

src/operators/unary.cc:95

MethodFlattenObj

src/operators/reshape.cc:67

include/nnet/derivator.h:18

include/core/object.h:43

MethodFullPrinterVisitor

include/nnet/Visitor/FullPrinterVisitor.h:11

include/nnet/expr.h:368

include/nnet/visitor.h:19

src/operators/G2BMM.cc:5

MethodG2bmmNode

include/nnet/routine.h:110

src/operators/GBMM.cc:5

MethodGatherBaseObj

include/operators/gather.h:12

MethodGatherElementsObj

src/operators/gather_elements.cc:5

MethodGatherObj

src/operators/gather.cc:5

include/nnet/routine.h:131

MethodGetTensorsVisitor

include/nnet/Visitor/GetTensorsVisitor.h:14

← previous next →601–700 of 2,429, ranked by callers