MCPcopy Create free account
hub / github.com/apache/tvm / PrimType

Class PrimType

python/tvm/ir/type.py:44–54  ·  view source on GitHub ↗

Primitive data type in the low level IR Parameters ---------- dtype : str The runtime data type relates to the primtype.

Source from the content-addressed store, hash-verified

42
43@tvm_ffi.register_object("ir.PrimType")
44class PrimType(Type):
45 """Primitive data type in the low level IR
46
47 Parameters
48 ----------
49 dtype : str
50 The runtime data type relates to the primtype.
51 """
52
53 def __init__(self, dtype):
54 self.__init_handle_by_constructor__(_ffi_api.PrimType, dtype)
55
56
57@tvm_ffi.register_object("ir.PointerType")

Callers 12

remote_viewMethod · 0.90
decl_bufferFunction · 0.90
visit_function_defFunction · 0.90
test_let_varFunction · 0.90
alloc_layoutFunction · 0.90
_build_expected_implFunction · 0.90
gemm_asyncFunction · 0.90
gemm_async_fnFunction · 0.90

Calls

no outgoing calls

Tested by 8

test_let_varFunction · 0.72
alloc_layoutFunction · 0.72
_build_expected_implFunction · 0.72
gemm_asyncFunction · 0.72
gemm_async_fnFunction · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…