[ET-VK][ez] Fix conv2d_pw shared memory buffer not having a constant size

[ET-VK][ez] Fix conv2d_pw shared memory buffer not having a constant size #4104