Skip to content

Commit f150e55

Browse files
internal developermryszt
authored andcommitted
Add meta for aten.poisson
Based on upstream commit pytorch@38f97ec Change-Id: I02ea48166ba96cf0c3ea841fa323e17b630fab08
1 parent 624ad9b commit f150e55

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

torch/_meta_registrations.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2044,6 +2044,12 @@ def meta_bernoulli_p(self, p=0.5, generator=None):
20442044
return torch.empty_like(self).contiguous()
20452045

20462046

2047+
@register_meta([aten.poisson.default, aten.poisson.out])
2048+
@out_wrapper()
2049+
def meta_poisson(self, generator=None):
2050+
return torch.empty_like(self)
2051+
2052+
20472053
@register_meta(aten._fused_moving_avg_obs_fq_helper.default)
20482054
def meta__fused_moving_avg_obs_fq_helper(
20492055
self,

0 commit comments

Comments
 (0)