The Op SpecifyShape can have a cpu or gpu input. It happen that this fct get called when the input first is a GpuArrayType. That is why I must add the check. op_lifter call the fct because the second input is a HostFromGpu, and he call that fct is any of the inputs is an HostFromGpu.