Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Archermmt/tvm_walk_through
/ functions
Functions
83 in github.com/Archermmt/tvm_walk_through
⨍
Functions
83
◇
Types & classes
2
↓ 54 callers
Method
visit
(self,expr,ref_type=None)
visualize.py:334
↓ 33 callers
Method
visualize
(self,mod,entry_name="main",path=None)
visualize.py:25
↓ 31 callers
Function
update_des
(kwargs,obj,key=None)
visualize.py:276
↓ 29 callers
Method
_append_parents
(self,node_des,depends)
visualize.py:792
↓ 28 callers
Method
_add_node
(self,node_ref,node_des,ref_type=None)
visualize.py:799
↓ 12 callers
Method
_add_node
(self,node_ref,node_des)
visualize.py:258
↓ 11 callers
Method
visit_elemwise
(self,expr,op_type,ref_type=None)
visualize.py:483
↓ 9 callers
Method
_update_depends
(self,node_des,refs)
visualize.py:179
↓ 8 callers
Method
_get_outputs
(self,ref,node_name=None)
visualize.py:196
↓ 7 callers
Function
array_des
(array)
utils.py:13
↓ 5 callers
Method
visit
(self,expr)
visualize.py:50
↓ 4 callers
Function
process_codegen
(target,optimize_mixed=False)
tvm_build/build_transform.py:14
↓ 3 callers
Function
cast_array
(array)
utils.py:5
↓ 3 callers
Function
compare_mod
(raw_mod,opt_mod,func_var=None,visual_name=None)
tvm_build/build_transform.py:42
↓ 3 callers
Function
get_pattern_kind
base utils
relay_optimize/visualize.cpp:70
↓ 3 callers
Method
visit_imm
(self,expr,optype,ref_type=None)
visualize.py:476
↓ 1 callers
Method
__init__
(self,name="relay_ir")
visualize.py:14
↓ 1 callers
Method
_get_node_des
(self,expr)
visualize.py:785
↓ 1 callers
Function
auto_optimize
(mod,target,params)
relay_optimize/demo.py:12
↓ 1 callers
Function
bind
(visualizer)
stmt_build/test_cases.py:121
↓ 1 callers
Function
bind
(visualizer)
stmt_optimize/demo.py:52
↓ 1 callers
Function
cache_read
(visualizer)
stmt_build/test_cases.py:7
↓ 1 callers
Function
calibrate_dataset
()
adv_quantize/demo.py:14
↓ 1 callers
Function
compute_inline
(visualizer)
stmt_build/test_cases.py:26
↓ 1 callers
Function
debug_optimize
(mod,target,params)
relay_optimize/test_index_graph.py:8
↓ 1 callers
Function
infer_type
A method to infer the type of a relay expression.
visualize.py:7
↓ 1 callers
Function
intrin_gemv
(m, l)
stmt_build/test_cases.py:87
↓ 1 callers
Method
reset
(self)
visualize.py:18
↓ 1 callers
Method
reset
(self)
visualize.py:305
↓ 1 callers
Function
simple
(visualizer)
stmt_optimize/demo.py:7
↓ 1 callers
Function
split
(visualizer)
stmt_build/test_cases.py:60
↓ 1 callers
Function
tensorize
(visualizer)
stmt_build/test_cases.py:78
↓ 1 callers
Function
test_lower_tvm_builtin
(func_var=None)
tvm_build/build_transform.py:116
↓ 1 callers
Function
test_make_packed_api
(func_var=None)
tvm_build/build_transform.py:78
↓ 1 callers
Function
test_split_host_device
(func_var=None)
tvm_build/build_transform.py:94
↓ 1 callers
Function
test_thread_sync
(func_var=None)
tvm_build/build_transform.py:66
↓ 1 callers
Function
vectorize
(visualizer)
stmt_optimize/demo.py:25
↓ 1 callers
Method
visit_allocate
(self,expr,ref_type=None)
visualize.py:492
↓ 1 callers
Method
visit_array
(self,expr,ref_type=None)
visualize.py:447
↓ 1 callers
Method
visit_assert
(self,expr,ref_type=None)
visualize.py:509
↓ 1 callers
Method
visit_attr
(self,expr,ref_type=None)
visualize.py:520
↓ 1 callers
Method
visit_buffer
(self,expr,ref_type=None)
visualize.py:537
↓ 1 callers
Method
visit_bufferload
(self,expr,ref_type=None)
visualize.py:545
↓ 1 callers
Method
visit_bufferrealize
(self,expr,ref_type=None)
visualize.py:555
↓ 1 callers
Method
visit_bufferstore
(self,expr,ref_type=None)
visualize.py:573
↓ 1 callers
Method
visit_call
(self,expr,ref_type=None)
visualize.py:591
↓ 1 callers
Method
visit_commreducer
(self,expr,ref_type=None)
visualize.py:606
↓ 1 callers
Method
visit_compute
(self,expr,ref_type=None)
visualize.py:438
↓ 1 callers
Method
visit_evalute
(self,expr,ref_type=None)
visualize.py:628
↓ 1 callers
Method
visit_for
(self,expr,ref_type=None)
visualize.py:636
↓ 1 callers
Method
visit_ifthenelse
(self,expr,ref_type=None)
visualize.py:653
↓ 1 callers
Method
visit_itervar
(self,expr,ref_type=None)
visualize.py:464
↓ 1 callers
Method
visit_let
(self,expr,ref_type=None)
visualize.py:669
↓ 1 callers
Method
visit_load
(self,expr,ref_type=None)
visualize.py:679
↓ 1 callers
Method
visit_placeholder
(self,expr,ref_type=None)
visualize.py:430
↓ 1 callers
Method
visit_primfunc
(self,expr,ref_type=None)
visualize.py:691
↓ 1 callers
Method
visit_producer
(self,expr,ref_type=None)
visualize.py:707
↓ 1 callers
Method
visit_ramp
(self,expr,ref_type=None)
visualize.py:717
↓ 1 callers
Method
visit_range
(self,expr,ref_type=None)
visualize.py:774
↓ 1 callers
Method
visit_reduce
(self,expr,ref_type=None)
visualize.py:729
↓ 1 callers
Method
visit_seq
(self,expr,ref_type=None)
visualize.py:748
↓ 1 callers
Method
visit_store
(self,expr,ref_type=None)
visualize.py:759
↓ 1 callers
Method
visit_tensor
(self,expr,ref_type=None)
visualize.py:426
↓ 1 callers
Method
visit_var
(self,expr,ref_type=None)
visualize.py:456
Method
__init__
(self,name="prim_expr",simple_mode=True)
visualize.py:300
Method
_tensor_des
(tensor)
visualize.py:31
Function
array_compare
(arrayA,arrayB,nameA="A",nameB="B",error=0.05)
utils.py:20
Function
check_optimize
(inter)
adv_interpreter/demo.py:13
Function
check_optimize
(mod,target,params)
adv_quantize/demo.py:20
Function
check_optimize
(mod,target,params)
adv_autotvm/demo.py:14
Function
check_optimize
(mod,params)
adv_tensorrt/demo.py:21
Function
check_optimize
(mod,target,params)
adv_virtualmachine/demo.py:14
Function
debug_optimize
(mod, params, disabled_pass, config)
dev_msir/torch_test.py:13
Function
debug_optimize
(mod,target,params)
relay_optimize/demo.py:18
Function
intrin_func
(ins, outs)
stmt_build/test_cases.py:96
Method
visit_call
(self,call)
visualize.py:106
Method
visit_constant
(self,const)
visualize.py:62
Method
visit_function
(self,func)
visualize.py:77
Method
visit_let
(self, let)
visualize.py:88
Method
visit_tuple
(self,tup)
visualize.py:69
Method
visit_var
(self,var)
visualize.py:54
Function
visualize
for DependencyGraph, add to DependencyGraph class
relay_optimize/visualize.cpp:2
Method
visualize
(self,expr,path=None,name='main',simple_mode=None)
visualize.py:312