MCPcopy Index your code
hub / github.com/apache/tvm / init

Method init

tests/python/tirx/test_parser_printer.py:743–753  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

741
742 @T.inline
743 def init(self):
744 self.Ta = self.Ta + T.float16(1)
745 self.Tb = self.Tb + T.float16(2)
746 self.idx.buffer[0] = T.int32(0)
747 self.idx = self.idx + T.int32(1)
748 self.inner_pool2 = self.inner_pool2 + T.float16(1)
749 T.evaluate(T.address_of(self.Ta))
750 T.evaluate(T.address_of(self.Tb))
751 T.evaluate(T.address_of(self.idx))
752 T.evaluate(T.address_of(self.inner_pool))
753 T.evaluate(T.address_of(self.inner_pool2))
754
755 @T.prim_func
756 def test():

Callers 15

testFunction · 0.95
matmulFunction · 0.45
tir_matmulMethod · 0.45
tir_matmulMethod · 0.45
test_convFunction · 0.45
matmulMethod · 0.45
sum_1dMethod · 0.45
fusedMethod · 0.45
tir_matmulMethod · 0.45
max_pool2d_openclMethod · 0.45
argmaxMethod · 0.45

Calls 1

evaluateMethod · 0.45

Tested by

no test coverage detected