(x)
| 281 | return f(builtin.Subtensor(items=items), src, *args) |
| 282 | |
| 283 | def _kernel_init(x): |
| 284 | k_1 = _subtensor(x, 0, None, c_1) |
| 285 | k_2 = _subtensor(x, 0, c_1, c_2) |
| 286 | k_3 = _subtensor(x, 0, c_2, c_3) |
| 287 | k_11 = f("-", _subtensor(k_1, 1, None, c_1), c_stg) |
| 288 | k_12_21 = f("-", _subtensor(k_1, 1, c_1, c_2), c_stg) |
| 289 | k_23_32 = f("+", _subtensor(k_2, 1, c_2, c_3), c_stg) |
| 290 | k_33 = f("+", _subtensor(k_3, 1, c_2, c_3), c_stg) |
| 291 | k_13 = _subtensor(k_1, 1, c_2, c_3) |
| 292 | k_22 = _subtensor(k_2, 1, c_1, c_2) |
| 293 | k_31 = _subtensor(k_3, 1, None, c_1) |
| 294 | nk_1 = f(builtin.Concat(axis=1), k_11, k_12_21, k_13,) |
| 295 | nk_2 = f(builtin.Concat(axis=1), k_12_21, k_22, k_23_32,) |
| 296 | nk_3 = f(builtin.Concat(axis=1), k_31, k_23_32, k_33,) |
| 297 | return f(builtin.Concat(axis=0), nk_1, nk_2, nk_3,) |
| 298 | |
| 299 | def _kernel_calc(k_e, k_n): |
| 300 | k1 = f("*", k_n, c_sub_alp) |
no test coverage detected