oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

Some Modules miss `dtype` property

Open MARD1NO opened this issue 3 years ago • 1 comments

In PyTorch,Conv2D module can pass dtype property to create different dtype parameter, but in OneFlow it miss this property.

MARD1NO avatar Jul 13 '22 01:07 MARD1NO

and device property

shangguanshiyuan avatar Jul 13 '22 01:07 shangguanshiyuan