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

Method full

tests/python/relax/test_vm_codegen_only.py:370–377  ·  view source on GitHub ↗
(T_full: T.Buffer((T.int64(4),), "float32"))

Source from the content-addressed store, hash-verified

368 class TestKillObject:
369 @T.prim_func(s_tir=True)
370 def full(T_full: T.Buffer((T.int64(4),), "float32")):
371 T.func_attr({"global_symbol": "full", "tirx.noalias": True})
372 for ax0 in range(T.int64(4)):
373 with T.sblock("T_full"):
374 v_ax0 = T.axis.spatial(T.int64(4), ax0)
375 T.reads()
376 T.writes(T_full[v_ax0])
377 T_full[v_ax0] = T.float32(0)
378
379 @T.prim_func(s_tir=True)
380 def full1(T_full: T.Buffer((T.int64(4),), "float32")):

Callers 15

_fillMethod · 0.45
_inplace_fillMethod · 0.45
_fullMethod · 0.45
_new_onesMethod · 0.45
_new_zerosMethod · 0.45
_onesMethod · 0.45
fullFunction · 0.45
_impl_v9Method · 0.45
_as_int64_tensorFunction · 0.45
convert_fillMethod · 0.45

Calls 1

spatialMethod · 0.80

Tested by

no test coverage detected