nn.AvgPool2d for fp16?

[Byung_Choi (Byung Il Choi)]

RuntimeError: “avg_pool2d_out_frame” not implemented for ‘Half’

I get this error for fp16 input. Does anyone know how to use AvgPool2d for half precision inputs?