(shape)
| 47 | def he_uniform(self, dev): |
| 48 | |
| 49 | def init(shape): |
| 50 | fan_in, _ = self.compute_fan(shape) |
| 51 | limit = np.sqrt(6 / fan_in) |
| 52 | return limit |
| 53 | |
| 54 | self.t1.to_device(dev) |
| 55 | initializer.he_uniform(self.t1) |
nothing calls this directly
no test coverage detected