Quantization and 3D convolution

Hello @OValery16, I believe the issue you are encountering is that you are calling te.thread_axis("threadIdx.z") multiple times. Instead, can you try creating the thread axis once with thread_z = te.thread_axis("threadIdx.y") and then use it like so: s[output].bind(s[output].fuse(tf, td), thread_z). I think you’ll also have to do this for threadIdx.x and threadIdx.y.