For the futur and consistency with the cpu version, use node.inputs[i].dtype, not float32.