MCPcopy Create free account
hub / github.com/pytorch/executorch / DropOutNode

Class DropOutNode

backends/example/example_operators/dropout.py:38–44  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

36
37@dataclass
38class DropOutNode(OpBase):
39 def __init__(self):
40 super().__init__(
41 # pattern=(torch.clone,),
42 pattern=(torch.nn.modules.dropout.Dropout,),
43 annotate_handle=_annotate_dropout,
44 )

Callers 1

ops.pyFile · 0.90

Calls

no outgoing calls

Tested by

no test coverage detected