MCPcopy Create free account

hub / github.com/InfiniTensor/InfiniTensor / functions

Functions2,429 in github.com/InfiniTensor/InfiniTensor

↓ 1 callersMethodgetWorkspaceObj
include/kunlun/kunlun_runtime.h:56
↓ 1 callersMethodget_data
include/core/perf_engine.h:44
↓ 1 callersFunctionget_kernel_attrs_str
src/utils/operator_utils.cc:114
↓ 1 callersFunctionget_model
(modelname)
examples/distributed/kunlun/run_pytorch.py:61
↓ 1 callersFunctionget_model
(modelname)
examples/distributed/bang/run_pytorch_mlu.py:46
↓ 1 callersFunctionget_model
(modelname)
examples/distributed/cuda/run_pytorch.py:42
↓ 1 callersFunctionget_type_name
include/nnet/dbg.h:157
↓ 1 callersMethodhasPadding
src/nnet/expr.cc:93
↓ 1 callersMethodhasSumVar
include/nnet/expr.h:246
↓ 1 callersFunctionhashLoopVar
src/nnet/Visitor/hashVisitor.cc:48
↓ 1 callersFunctionhashSumVar
src/nnet/Visitor/hashVisitor.cc:52
↓ 1 callersMethodheapAlloc
src/core/lazy_allocator.cc:119
↓ 1 callersFunctionidx2Pos
src/kernels/cpu/transpose.cc:6
↓ 1 callersMethodinferShape
src/core/operator.cc:82
↓ 1 callersMethodinit
src/core/lazy_allocator.cc:38
↓ 1 callersFunctioninit_graph_builder
src/ffi/ffi_infinitensor.cc:435
↓ 1 callersMethodinstanceNormalization
src/core/graph_handler.cc:139
↓ 1 callersFunctionisColorizedOutputEnabled
include/nnet/dbg.h:75
↓ 1 callersMethodisHead
include/core/graph_match.h:20
↓ 1 callersMethodisInput
include/core/tensor.h:44
↓ 1 callersMethodisInputFromOutside
include/core/graph_match.h:14
↓ 1 callersMethodisOutput
include/core/tensor.h:45
↓ 1 callersMethodisOutput2Outside
include/core/graph_match.h:17
↓ 1 callersMethodisSuccessful
src/nnet/Pass/Rule4StageMerging.cc:11
↓ 1 callersMethodisSwapable
src/nnet/expr.cc:239
↓ 1 callersMethodisTail
include/core/graph_match.h:28
↓ 1 callersMethodis_partial
(self)
examples/distributed/placement.py:17
↓ 1 callersMethodlayerNormalization
src/core/graph_handler.cc:122
↓ 1 callersMethodleakyRelu
src/core/graph_handler.cc:251
↓ 1 callersMethodloadPerfEngineData
src/core/perf_engine.cc:15
↓ 1 callersFunctionloadTensorData
src/utils/dataloader.cc:48
↓ 1 callersFunctionload_inputs
(stub, world_size=1, rank=0)
examples/distributed/ascend_launch.py:136
↓ 1 callersFunctionload_inputs
(stub: OnnxStub)
examples/distributed/kunlun/kunlun_launch.py:133
↓ 1 callersFunctionload_inputs
(stub, world_size=1, rank=0)
examples/distributed/bang/bang_launch.py:70
↓ 1 callersFunctionlocate_index
src/utils/operator_utils.cc:69
↓ 1 callersMethodlrn
src/core/graph_handler.cc:661
↓ 1 callersFunctionmain
()
examples/distributed/ascend_launch.py:158
↓ 1 callersFunctionmain
()
examples/distributed/kunlun/kunlun_launch.py:238
↓ 1 callersFunctionmain
()
examples/distributed/kunlun/run_pytorch.py:225
↓ 1 callersFunctionmain
()
examples/distributed/bang/run_pytorch_mlu.py:227
↓ 1 callersFunctionmain
()
examples/distributed/bang/bang_launch.py:145
↓ 1 callersFunctionmain
()
examples/distributed/cuda/launch_kvcache.py:207
↓ 1 callersFunctionmain
()
examples/distributed/cuda/cuda_launch.py:120
↓ 1 callersFunctionmain
()
examples/distributed/cuda/run_pytorch.py:167
↓ 1 callersMethodname_op
(self, op: backend.Operator)
pyinfinitensor/src/pyinfinitensor/onnx.py:1154
↓ 1 callersFunctionnp_assert
(base, test, rtol=1e-2, atol=1e-1)
examples/distributed/kunlun/kunlun_launch.py:84
↓ 1 callersMethodoperators
include/core/graph_handler.h:26
↓ 1 callersMethodpRelu
src/core/graph_handler.cc:241
↓ 1 callersFunctionparseActType
include/kunlun/kunlun_act_type.h:6
↓ 1 callersFunctionparse_args
()
examples/distributed/ascend_launch.py:13
↓ 1 callersFunctionparse_args
()
examples/distributed/kunlun/kunlun_launch.py:17
↓ 1 callersFunctionparse_args
()
examples/distributed/kunlun/run_pytorch.py:17
↓ 1 callersFunctionparse_args
()
examples/distributed/bang/run_pytorch_mlu.py:17
↓ 1 callersFunctionparse_args
()
examples/distributed/bang/bang_launch.py:15
↓ 1 callersFunctionparse_args
()
examples/distributed/cuda/launch_kvcache.py:15
↓ 1 callersFunctionparse_args
()
examples/distributed/cuda/cuda_launch.py:13
↓ 1 callersFunctionparse_args
()
examples/distributed/cuda/run_pytorch.py:13
↓ 1 callersFunctionparse_attribute
(node: NodeProto, attrs: Dict[str, Any] = dict())
examples/distributed/parallel.py:12
↓ 1 callersFunctionperf_infinitensor
(stub: OnnxStub)
examples/distributed/kunlun/kunlun_launch.py:171
↓ 1 callersMethodprefix
include/nnet/dbg.h:119
↓ 1 callersMethodpush_input
( self, tensor: backend.Tensor, init: Optional[TensorProto] )
pyinfinitensor/src/pyinfinitensor/onnx.py:1171
↓ 1 callersMethodpush_output
(self, name: str, tensor: backend.Tensor)
pyinfinitensor/src/pyinfinitensor/onnx.py:1161
↓ 1 callersMethodrecv
src/core/graph_handler.cc:592
↓ 1 callersFunctionref_addr_less
include/nnet/ref.h:72
↓ 1 callersFunctionref_value_less
include/nnet/ref.h:76
↓ 1 callersFunctionregister_operator_timer
src/ffi/ffi_infinitensor.cc:47
↓ 1 callersMethodremovePredecessors
src/core/operator.cc:15
↓ 1 callersMethodremoveSuccessors
src/core/operator.cc:24
↓ 1 callersMethodremoveTarget
include/core/tensor_base.h:60
↓ 1 callersMethodreorder
src/nnet/dlt.cc:127
↓ 1 callersMethodreplaceConnection
src/core/graph.cc:339
↓ 1 callersMethodreplaceInput
src/core/operator.cc:33
↓ 1 callersFunctionreplace_onnx_with_attention_op
()
examples/python/llama_kvcache_inference.py:58
↓ 1 callersMethodrule4StageMerging
src/nnet/Pass/Rule4StageMerging.cc:17
↓ 1 callersMethodrule5RangeRelaxation
src/nnet/derivator.cc:348
↓ 1 callersMethodrun
src/core/search_engine.cc:31
↓ 1 callersMethodrun
include/core/graph_handler.h:148
↓ 1 callersFunctionrun_cifar_train_and_infer
()
examples/python/paddle_densenet.py:9
↓ 1 callersFunctionrun_cifar_train_and_infer
()
examples/python/paddle_resnet.py:10
↓ 1 callersFunctionrun_cifar_train_and_infer
()
examples/python/paddle_inception.py:8
↓ 1 callersFunctionrun_model
(model, runtime, world_size=1, rank=0, n=10)
examples/distributed/ascend_launch.py:47
↓ 1 callersFunctionrun_model
(model, runtime, world_size=1, rank=0, n=10, data_type="default")
examples/distributed/bang/bang_launch.py:51
↓ 1 callersFunctionrun_pytorch
(torch_model, voc_size, batchsize, len, model_name)
examples/distributed/kunlun/run_pytorch.py:80
↓ 1 callersFunctionrun_pytorch
(torch_model, voc_size, batchsize, len, dtype="fp32")
examples/distributed/bang/run_pytorch_mlu.py:69
↓ 1 callersFunctionrun_pytorch
(torch_model, voc_size, batchsize, len)
examples/distributed/cuda/run_pytorch.py:59
↓ 1 callersMethodsavePerfEngineData
src/core/perf_engine.cc:7
↓ 1 callersFunctionsaveTensorData
src/utils/dataloader.cc:11
↓ 1 callersMethodsend
src/core/graph_handler.cc:578
↓ 1 callersMethodsetLhs
include/nnet/expr.h:304
↓ 1 callersMethodsetLoopIterator
include/nnet/expr.h:255
↓ 1 callersMethodsetMemPool
src/core/lazy_allocator.cc:50
↓ 1 callersMethodsetMergeStageWithCalc
src/nnet/Pass/Rule4StageMerging.cc:13
↓ 1 callersMethodsetRhs
include/nnet/expr.h:305
↓ 1 callersMethodsetSource
include/core/tensor_base.h:59
↓ 1 callersMethodsetToNaiveMembound
src/nnet/nmutator.cc:25
↓ 1 callersMethodsetVarRange
include/nnet/expr.h:277
↓ 1 callersMethodset_data
include/core/perf_engine.h:45
↓ 1 callersMethodshape_infer
src/core/graph.cc:139
↓ 1 callersFunctionshard_concat
(node: NodeProto)
examples/distributed/parallel_opt.py:61
↓ 1 callersFunctionshard_gemm
(node: NodeProto, groups: int = 1)
examples/distributed/parallel_opt.py:46
← previousnext →501–600 of 2,429, ranked by callers