Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/InfiniTensor/InfiniTensor
/ functions
Functions
2,429 in github.com/InfiniTensor/InfiniTensor
⨍
Functions
2,429
◇
Types & classes
632
↓ 1 callers
Method
getWorkspaceObj
include/kunlun/kunlun_runtime.h:56
↓ 1 callers
Method
get_data
include/core/perf_engine.h:44
↓ 1 callers
Function
get_kernel_attrs_str
src/utils/operator_utils.cc:114
↓ 1 callers
Function
get_model
(modelname)
examples/distributed/kunlun/run_pytorch.py:61
↓ 1 callers
Function
get_model
(modelname)
examples/distributed/bang/run_pytorch_mlu.py:46
↓ 1 callers
Function
get_model
(modelname)
examples/distributed/cuda/run_pytorch.py:42
↓ 1 callers
Function
get_type_name
include/nnet/dbg.h:157
↓ 1 callers
Method
hasPadding
src/nnet/expr.cc:93
↓ 1 callers
Method
hasSumVar
include/nnet/expr.h:246
↓ 1 callers
Function
hashLoopVar
src/nnet/Visitor/hashVisitor.cc:48
↓ 1 callers
Function
hashSumVar
src/nnet/Visitor/hashVisitor.cc:52
↓ 1 callers
Method
heapAlloc
src/core/lazy_allocator.cc:119
↓ 1 callers
Function
idx2Pos
src/kernels/cpu/transpose.cc:6
↓ 1 callers
Method
inferShape
src/core/operator.cc:82
↓ 1 callers
Method
init
src/core/lazy_allocator.cc:38
↓ 1 callers
Function
init_graph_builder
src/ffi/ffi_infinitensor.cc:435
↓ 1 callers
Method
instanceNormalization
src/core/graph_handler.cc:139
↓ 1 callers
Function
isColorizedOutputEnabled
include/nnet/dbg.h:75
↓ 1 callers
Method
isHead
include/core/graph_match.h:20
↓ 1 callers
Method
isInput
include/core/tensor.h:44
↓ 1 callers
Method
isInputFromOutside
include/core/graph_match.h:14
↓ 1 callers
Method
isOutput
include/core/tensor.h:45
↓ 1 callers
Method
isOutput2Outside
include/core/graph_match.h:17
↓ 1 callers
Method
isSuccessful
src/nnet/Pass/Rule4StageMerging.cc:11
↓ 1 callers
Method
isSwapable
src/nnet/expr.cc:239
↓ 1 callers
Method
isTail
include/core/graph_match.h:28
↓ 1 callers
Method
is_partial
(self)
examples/distributed/placement.py:17
↓ 1 callers
Method
layerNormalization
src/core/graph_handler.cc:122
↓ 1 callers
Method
leakyRelu
src/core/graph_handler.cc:251
↓ 1 callers
Method
loadPerfEngineData
src/core/perf_engine.cc:15
↓ 1 callers
Function
loadTensorData
src/utils/dataloader.cc:48
↓ 1 callers
Function
load_inputs
(stub, world_size=1, rank=0)
examples/distributed/ascend_launch.py:136
↓ 1 callers
Function
load_inputs
(stub: OnnxStub)
examples/distributed/kunlun/kunlun_launch.py:133
↓ 1 callers
Function
load_inputs
(stub, world_size=1, rank=0)
examples/distributed/bang/bang_launch.py:70
↓ 1 callers
Function
locate_index
src/utils/operator_utils.cc:69
↓ 1 callers
Method
lrn
src/core/graph_handler.cc:661
↓ 1 callers
Function
main
()
examples/distributed/ascend_launch.py:158
↓ 1 callers
Function
main
()
examples/distributed/kunlun/kunlun_launch.py:238
↓ 1 callers
Function
main
()
examples/distributed/kunlun/run_pytorch.py:225
↓ 1 callers
Function
main
()
examples/distributed/bang/run_pytorch_mlu.py:227
↓ 1 callers
Function
main
()
examples/distributed/bang/bang_launch.py:145
↓ 1 callers
Function
main
()
examples/distributed/cuda/launch_kvcache.py:207
↓ 1 callers
Function
main
()
examples/distributed/cuda/cuda_launch.py:120
↓ 1 callers
Function
main
()
examples/distributed/cuda/run_pytorch.py:167
↓ 1 callers
Method
name_op
(self, op: backend.Operator)
pyinfinitensor/src/pyinfinitensor/onnx.py:1154
↓ 1 callers
Function
np_assert
(base, test, rtol=1e-2, atol=1e-1)
examples/distributed/kunlun/kunlun_launch.py:84
↓ 1 callers
Method
operators
include/core/graph_handler.h:26
↓ 1 callers
Method
pRelu
src/core/graph_handler.cc:241
↓ 1 callers
Function
parseActType
include/kunlun/kunlun_act_type.h:6
↓ 1 callers
Function
parse_args
()
examples/distributed/ascend_launch.py:13
↓ 1 callers
Function
parse_args
()
examples/distributed/kunlun/kunlun_launch.py:17
↓ 1 callers
Function
parse_args
()
examples/distributed/kunlun/run_pytorch.py:17
↓ 1 callers
Function
parse_args
()
examples/distributed/bang/run_pytorch_mlu.py:17
↓ 1 callers
Function
parse_args
()
examples/distributed/bang/bang_launch.py:15
↓ 1 callers
Function
parse_args
()
examples/distributed/cuda/launch_kvcache.py:15
↓ 1 callers
Function
parse_args
()
examples/distributed/cuda/cuda_launch.py:13
↓ 1 callers
Function
parse_args
()
examples/distributed/cuda/run_pytorch.py:13
↓ 1 callers
Function
parse_attribute
(node: NodeProto, attrs: Dict[str, Any] = dict())
examples/distributed/parallel.py:12
↓ 1 callers
Function
perf_infinitensor
(stub: OnnxStub)
examples/distributed/kunlun/kunlun_launch.py:171
↓ 1 callers
Method
prefix
include/nnet/dbg.h:119
↓ 1 callers
Method
push_input
( self, tensor: backend.Tensor, init: Optional[TensorProto] )
pyinfinitensor/src/pyinfinitensor/onnx.py:1171
↓ 1 callers
Method
push_output
(self, name: str, tensor: backend.Tensor)
pyinfinitensor/src/pyinfinitensor/onnx.py:1161
↓ 1 callers
Method
recv
src/core/graph_handler.cc:592
↓ 1 callers
Function
ref_addr_less
include/nnet/ref.h:72
↓ 1 callers
Function
ref_value_less
include/nnet/ref.h:76
↓ 1 callers
Function
register_operator_timer
src/ffi/ffi_infinitensor.cc:47
↓ 1 callers
Method
removePredecessors
src/core/operator.cc:15
↓ 1 callers
Method
removeSuccessors
src/core/operator.cc:24
↓ 1 callers
Method
removeTarget
include/core/tensor_base.h:60
↓ 1 callers
Method
reorder
src/nnet/dlt.cc:127
↓ 1 callers
Method
replaceConnection
src/core/graph.cc:339
↓ 1 callers
Method
replaceInput
src/core/operator.cc:33
↓ 1 callers
Function
replace_onnx_with_attention_op
()
examples/python/llama_kvcache_inference.py:58
↓ 1 callers
Method
rule4StageMerging
src/nnet/Pass/Rule4StageMerging.cc:17
↓ 1 callers
Method
rule5RangeRelaxation
src/nnet/derivator.cc:348
↓ 1 callers
Method
run
src/core/search_engine.cc:31
↓ 1 callers
Method
run
include/core/graph_handler.h:148
↓ 1 callers
Function
run_cifar_train_and_infer
()
examples/python/paddle_densenet.py:9
↓ 1 callers
Function
run_cifar_train_and_infer
()
examples/python/paddle_resnet.py:10
↓ 1 callers
Function
run_cifar_train_and_infer
()
examples/python/paddle_inception.py:8
↓ 1 callers
Function
run_model
(model, runtime, world_size=1, rank=0, n=10)
examples/distributed/ascend_launch.py:47
↓ 1 callers
Function
run_model
(model, runtime, world_size=1, rank=0, n=10, data_type="default")
examples/distributed/bang/bang_launch.py:51
↓ 1 callers
Function
run_pytorch
(torch_model, voc_size, batchsize, len, model_name)
examples/distributed/kunlun/run_pytorch.py:80
↓ 1 callers
Function
run_pytorch
(torch_model, voc_size, batchsize, len, dtype="fp32")
examples/distributed/bang/run_pytorch_mlu.py:69
↓ 1 callers
Function
run_pytorch
(torch_model, voc_size, batchsize, len)
examples/distributed/cuda/run_pytorch.py:59
↓ 1 callers
Method
savePerfEngineData
src/core/perf_engine.cc:7
↓ 1 callers
Function
saveTensorData
src/utils/dataloader.cc:11
↓ 1 callers
Method
send
src/core/graph_handler.cc:578
↓ 1 callers
Method
setLhs
include/nnet/expr.h:304
↓ 1 callers
Method
setLoopIterator
include/nnet/expr.h:255
↓ 1 callers
Method
setMemPool
src/core/lazy_allocator.cc:50
↓ 1 callers
Method
setMergeStageWithCalc
src/nnet/Pass/Rule4StageMerging.cc:13
↓ 1 callers
Method
setRhs
include/nnet/expr.h:305
↓ 1 callers
Method
setSource
include/core/tensor_base.h:59
↓ 1 callers
Method
setToNaiveMembound
src/nnet/nmutator.cc:25
↓ 1 callers
Method
setVarRange
include/nnet/expr.h:277
↓ 1 callers
Method
set_data
include/core/perf_engine.h:45
↓ 1 callers
Method
shape_infer
src/core/graph.cc:139
↓ 1 callers
Function
shard_concat
(node: NodeProto)
examples/distributed/parallel_opt.py:61
↓ 1 callers
Function
shard_gemm
(node: NodeProto, groups: int = 1)
examples/distributed/parallel_opt.py:46
← previous
next →
501–600 of 2,429, ranked by callers