| import taichi as ti | |
| import taichi.math as tm | |
| """ @ti.kernel | |
| def costvol_out(tenOne: ti.types.ndarray(), tltOne: ti.types.ndarray(), tenTwo: ti.types.ndarray(), tenOut: ti.types.ndarray()): | |
| N, C, H, W = tenOut.shape | |
| for i, ch, y, x in ti.ndrange(N, C, H, W): | |
| for intValue in range(tenOne.shape[1]): | |
| tltOne[intValue] = tenOne[i, intValue, y, x] | |
| tenOut_ch = 0 | |
| for intOy in range(y - 4, y + 4 + 1): | |
| for intOx in range(x - 4, x + 4 + 1): | |
| point = tm.ivec2(intOx, intOy) | |
| fltValue = 0.0 | |
| for intValue in range(ch): | |
| if (point.y >= 0) and (point.y < H) and (point.x >= 0) and (point.x < W): | |
| fltValue += ti.abs(tltOne[intValue] - tenTwo[i, intValue, point.y, point.x]) | |
| else: | |
| fltValue += ti.abs(tltOne[intValue]) | |
| tenOut[i, tenOut_ch, y, x] = fltValue / tenOne.shape[1] | |
| tenOut_ch += 1 """ | |
| def worker_interface(op_name, tensors): | |
| raise NotImplementedError(op_name) | |