Changing device for the input tensor was not added in the previous MR (!130 (merged)) which is added in this MR.