After a bit of reflexion (and looking a bit up in the code and reading the commit message) I'm pretty sure this meant to go on the GpuAlloc op.  But it's on the GpuReshape op now.