I would like to know if Halide would be of any good for the following problem.
Consider an array X of reals of size m x n x 3.
I have a loop of the following f