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

↓ 11 callersMethodsetOutputs

include/core/graph_match.h:11

↓ 10 callersFunctionbroadcastShape

src/utils/operator_utils.cc:125

↓ 10 callersMethodgetInputsFromOutside

include/core/graph_match.h:12

↓ 10 callersMethodgetNumOutputDims

src/nnet/expr.cc:126

↓ 10 callersMethodgetPerfData

* @brief Get the Perf Data object * * @return PerfRecord nullptr if no record is fnoud. */

include/core/perf_engine.h:32

↓ 10 callersMethodgetRange

src/nnet/expr.cc:140

↓ 10 callersMethodpush_data_input

( self, node_name: str, attr_name: str, elem_t

pyinfinitensor/src/pyinfinitensor/onnx.py:1191

↓ 10 callersMethodtoString

src/core/graph.cc:68

↓ 9 callersMethodanalyzeExpr

src/nnet/iterator_table.cc:9

↓ 9 callersMethoddealloc

include/core/runtime.h:127

↓ 9 callersMethodgetMode

include/operators/det.h:15

↓ 9 callersMethodgetTransA

include/operators/matmul.h:60

↓ 9 callersMethodgetTransB

include/operators/matmul.h:61

↓ 9 callersMethodhash

src/core/operator.cc:52

↓ 9 callersFunctionroutineTypeToId

include/nnet/expr.h:45

↓ 9 callersMethodsetDumpFirstSuccess

src/nnet/derivator.cc:483

↓ 9 callersMethodsplit

src/nnet/dlt.cc:121

↓ 8 callersMethodcheckRangeOp

src/nnet/Visitor/CheckOOBVisitor.cc:28

↓ 8 callersMethodcopyBlobFromCPU

src/core/runtime.cc:148

↓ 8 callersMethodflatten

src/core/graph_handler.cc:280

↓ 8 callersMethodgetArgs

src/nnet/routine.cc:79

↓ 8 callersMethodgetAxes

include/operators/squeeze.h:32

↓ 8 callersMethodgetNCHWRS

include/operators/pooling.h:60

↓ 8 callersMethodgetTargetOp

src/nnet/derivator.cc:456

↓ 8 callersMethodgetWorldSize

include/operators/all_gather.h:34

↓ 8 callersMethodhasPaddings

src/nnet/expr.cc:410

↓ 8 callersMethodmerge

src/nnet/dlt.cc:124

↓ 8 callersMethodsync

src/kunlun/kunlun_runtime.cc:56

↓ 7 callersMethodcompute

src/kernels/cpu/unary.cc:282

↓ 7 callersMethoddata_malloc

include/core/graph_handler.h:137

↓ 7 callersFunctiongenhash

include/nnet/common.h:57

↓ 7 callersMethodget

include/nnet/permutation.h:29

↓ 7 callersMethodgetDataBlob

include/core/tensor_base.h:36

↓ 7 callersMethodgetKernel

include/core/kernel.h:157

↓ 7 callersMethodgetPads

include/operators/pad.h:34

↓ 7 callersMethodgetSearchedMaxDepth

include/nnet/derivator.h:84

↓ 7 callersMethodisCpu

include/core/runtime.h:72

↓ 7 callersMethodnumOutputs

include/operators/lrn.h:16

↓ 7 callersMethodreplace

src/nnet/Visitor/ReplaceNodeMutator.cc:16

↓ 7 callersFunctionserializeVec

src/nnet/expr.cc:6

↓ 6 callersMethodcopyBlobToCPU

src/core/runtime.cc:153

↓ 6 callersMethoddeserialize

src/nnet/Visitor/Serializer.cc:132

↓ 6 callersMethodgetAlpha

include/operators/unary.h:59

↓ 6 callersMethodgetMax

include/operators/unary.h:40

↓ 6 callersMethodgetMin

include/operators/unary.h:39

↓ 6 callersMethodgetNcclComm

Get the actual ncclComm_t

include/cuda/nccl_communicator.h:54

↓ 6 callersMethodgetObjectPtr

include/nnet/expr.h:352

↓ 6 callersMethodgetOpType

src/kernels/cuda/unary.cc:190

↓ 6 callersMethodgetOpType

src/kernels/cuda/element_wise.cc:101

↓ 6 callersMethodgetOutDType

include/core/operator.h:93

↓ 6 callersMethodgetPassMode

src/nnet/derivator.cc:527

↓ 6 callersMethodgetPermute

include/operators/transpose.h:15

↓ 6 callersMethodgetRoi

include/operators/resize.h:136

↓ 6 callersFunctioninfer_broadcast

src/utils/operator_utils.cc:6

↓ 6 callersMethodisWeight

include/core/tensor.h:43

↓ 6 callersMethodmatch

src/nnet/Visitor/ReplaceVariable.cc:39

↓ 6 callersFunctionmatchMatmul

test/nnet/test_matchMatmul.cc:10

↓ 6 callersFunctionparallel_model

(model: ModelProto, tp_world_size: int = 1, tp_rank: int = 0)

examples/distributed/parallel_opt.py:9

↓ 6 callersMethodsetPaddings

src/nnet/expr.cc:408

↓ 6 callersMethodsetPerfData

include/core/perf_engine.h:40

↓ 6 callersMethodsetSumIterator

include/nnet/expr.h:258

↓ 6 callersMethodsync

src/bang/bang_runtime.cc:63

↓ 5 callersMethodbuildTable

src/nnet/iterator_table.cc:20

↓ 5 callersMethodbuildTableWithDefaultMap

src/nnet/iterator_table.cc:88

↓ 5 callersMethodcopyBlobInsideRuntime

src/core/runtime.cc:158

↓ 5 callersMethodfree

src/core/lazy_allocator.cc:130

↓ 5 callersMethodgetAlignedSize

src/core/lazy_allocator.cc:207

↓ 5 callersMethodgetB

include/operators/GBMM.h:45

↓ 5 callersMethodgetBangQueue

include/bang/bang_runtime.h:81

↓ 5 callersMethodgetCeilMode

include/operators/pooling.h:57

↓ 5 callersMethodgetCoordinateTransMode

include/operators/resize.h:128

↓ 5 callersMethodgetDims

include/operators/reshape.h:34

↓ 5 callersMethodgetKeepDims

include/operators/reduce.h:37

↓ 5 callersMethodgetOutputSize

src/nnet/expr.cc:435

↓ 5 callersMethodgetOutputs2Outside

include/core/graph_match.h:13

↓ 5 callersMethodgetStarts

include/operators/slice.h:40

↓ 5 callersFunctionhashAppend

include/core/hash.h:5

↓ 5 callersMethodinterpret

src/nnet/Visitor/Interpreter.cc:142

↓ 5 callersMethodnext

src/nnet/permutation.cc:15

↓ 5 callersMethodpad

src/core/graph_handler.cc:499

↓ 5 callersMethodsetSearchState

src/nnet/derivator.cc:462

↓ 5 callersMethodsetSummand

include/nnet/expr.h:254

↓ 5 callersFunctiontestMatmulCuda

test/kernels/cuda/test_cuda_matmul.cc:13

↓ 5 callersFunctiontestMatmulKUNLUN

test/kernels/kunlun/test_kunlun_matmul.cc:51

↓ 4 callersFunctionTEST

test/nnet/test_dlt.cc:8

↓ 4 callersMethodapply

src/nnet/dlt.cc:8

↓ 4 callersMethodavgPool

src/core/graph_handler.cc:179

↓ 4 callersMethodbuild

(self, name: str)

pyinfinitensor/src/pyinfinitensor/onnx.py:1209

↓ 4 callersMethodconv

src/core/graph_handler.cc:55

↓ 4 callersMethodcopyData

src/core/tensor.cc:168

↓ 4 callersMethodcpuTypeInt

include/core/data_type.h:69

↓ 4 callersMethodcublasHandle

include/cuda/cuda_runtime.h:64

↓ 4 callersMethodgetBMNKTransAB

include/operators/matmul.h:59

↓ 4 callersMethodgetCnclComm

Get the actual cnclComm_t

include/bang/cncl_communicator.h:70

↓ 4 callersMethodgetComputeFunc

Get the optimal computing function according to the key

include/core/kernel.h:84

↓ 4 callersMethodgetDestinationRank

include/operators/send.h:31

↓ 4 callersMethodgetHcclComm

Get the actual ncclComm_t

include/ascend/hccl_communicator.h:58

↓ 4 callersMethodgetM

include/operators/GBMM.h:46

↓ 4 callersMethodgetN

include/operators/GBMM.h:48

↓ 4 callersMethodgetNumGroups

include/operators/conv.h:146

← previous next →101–200 of 2,429, ranked by callers