Hi Richard, why does flattening parameters cause an error on the GPU but not the CPU?
Also, what other types of operations throw inplace errors on the GPU but not CPU? I have the same problem as OP, but I do not flatten my parameters.