python/dgl/nn/pytorch/conv/densesageconv.py · dde5cf5d02188c7f493caef32d4ed59fc1805a2c · OpenDAS / dgl · GitLab

Find file Blame History Permalink

[Bug] Fix NN modules crashing with non-FP32 inputs (reopen #4262) (#4829) · dde5cf5d
Xin Yao authored Jan 06, 2023
```
* fix fp16 in nn modules

* fix lint

* remove the example
Co-authored-by: Quan Gan <coin2028@hotmail.com>
```
dde5cf5d

densesageconv.py 4.57 KB