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

Method concatenate

tests/python/relax/test_transform_fuse_tir.py:1049–1069  ·  view source on GitHub ↗
(
            rxplaceholder: T.Buffer((T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"),
            rxplaceholder_1: T.Buffer(
                (T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"
            ),
            T_concat: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"),
        )

Source from the content-addressed store, hash-verified

1047 class Module:
1048 @T.prim_func(private=True, s_tir=True)
1049 def concatenate(
1050 rxplaceholder: T.Buffer((T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"),
1051 rxplaceholder_1: T.Buffer(
1052 (T.int64(1), T.int64(4), T.int64(64), T.int64(64)), "float32"
1053 ),
1054 T_concat: T.Buffer((T.int64(2), T.int64(4), T.int64(64), T.int64(64)), "float32"),
1055 ):
1056 T.func_attr({"op_pattern": 2, "tirx.noalias": True})
1057 for ax0, ax1, ax2, ax3 in T.grid(T.int64(2), T.int64(4), T.int64(64), T.int64(64)):
1058 with T.sblock("T_concat"):
1059 v_ax0, v_ax1, v_ax2, v_ax3 = T.axis.remap("SSSS", [ax0, ax1, ax2, ax3])
1060 T.reads(
1061 rxplaceholder_1[v_ax0 - T.int64(1), v_ax1, v_ax2, v_ax3],
1062 rxplaceholder[v_ax0, v_ax1, v_ax2, v_ax3],
1063 )
1064 T.writes(T_concat[v_ax0, v_ax1, v_ax2, v_ax3])
1065 T_concat[v_ax0, v_ax1, v_ax2, v_ax3] = T.if_then_else(
1066 T.int64(1) <= v_ax0,
1067 rxplaceholder_1[v_ax0 - T.int64(1), v_ax1, v_ax2, v_ax3],
1068 rxplaceholder[v_ax0, v_ax1, v_ax2, v_ax3],
1069 )
1070
1071 @T.prim_func(private=True, s_tir=True)
1072 def transpose2(

Callers 15

autopadFunction · 0.45
_impl_v13Method · 0.45
_impl_v9Method · 0.45
train_fullMethod · 0.45
predict_incrementalMethod · 0.45
__init__Method · 0.45
updateMethod · 0.45
concatenateFunction · 0.45
conv3d_ndhwc_pythonFunction · 0.45
conv2d_nchw_pythonFunction · 0.45
conv2d_nhwc_pythonFunction · 0.45
group_conv1d_ncw_pythonFunction · 0.45

Calls 1

remapMethod · 0.80

Tested by

no test coverage detected