def METHOD_NAME(self):
    graph_ir = """
    graph(%1 : Float(2, 3)):
      %2 : float = prim::Constant[value=0.0]()
      %training : bool = prim::Constant[value=1]()
      %3 : Tensor, %4 : Tensor = aten::native_dropout(%1, %2, %training)
      return (%3, %4)
    """
    a = torch.randn(2, 3)
    self.run_test(graph_ir, (a,))