Lines Matching refs:data_size
69 data_size = data.shape[axis]
75 tvm.if_then_else(i < buflen - data_size,
76 buffer[i + data_size],
77 data[i - buflen + data_size]),
83 tvm.if_then_else(i < buflen - data_size,
84 buffer[i + data_size, j],
85 data[i - buflen + data_size, j]),
90 tvm.if_then_else(j < buflen - data_size,
91 buffer[i, j + data_size],
92 data[i, j - buflen + data_size]),
99 tvm.if_then_else(i < buflen - data_size,
100 buffer[i + data_size, j, k],
101 data[i - buflen + data_size, j, k]),
106 tvm.if_then_else(j < buflen - data_size,
107 buffer[i, j + data_size, k],
108 data[i, j - buflen + data_size, k]),
113 tvm.if_then_else(k < buflen - data_size,
114 buffer[i, j, k + data_size],
115 data[i, j, k - buflen + data_size]),
122 tvm.if_then_else(i < buflen - data_size,
123 buffer[i + data_size, j, k, l],
124 data[i - buflen + data_size, j, k, l]),
129 tvm.if_then_else(j < buflen - data_size,
130 buffer[i, j + data_size, k, l],
131 data[i, j - buflen + data_size, k, l]),
136 tvm.if_then_else(k < buflen - data_size,
137 buffer[i, j, k + data_size, l],
138 data[i, j, k - buflen + data_size, l]),
143 tvm.if_then_else(l < buflen - data_size,
144 buffer[i, j, k, l + data_size],
145 data[i, j, k, l - buflen + data_size]),