formatting
This commit is contained in:
parent
b62c88600f
commit
e3e41a7cef
1 changed files with 1 additions and 0 deletions
|
|
@ -63,6 +63,7 @@ def _seeded_dropout(
|
|||
tl.store(output_ptr + off2, o2, mask = mask2)
|
||||
tl.store(output_ptr + off3, o3, mask = mask3)
|
||||
|
||||
pass
|
||||
|
||||
def seeded_dropout(
|
||||
x: torch.Tensor,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue