Add methods to torch._C pyi (pytorch#78757)

Add python bound methods in `Block`, `Node`, `Graph` and `Value` to `torch._C.__init__.pyi` to enable proper type hints in editors. Pull Request resolved: pytorch#78757 Approved by: https://github.com/eellison, https://github.com/BowenBao

Author: justinchuby

torch/_C/__init__.pyi.in

@@ -457,40 +457,80 @@ class Value:
  def type(self)-> JitType: ...
  def setType(self, t: JitType) -> Value: ...
  def debugName(self) -> str: ...
+ def requires_grad(self) -> _bool: ...
  ...
 
 # Defined in torch/csrc/jit/ir/ir.h
 class Block:
  def inputs(self) -> List[Value]: ...
  def outputs(self) -> List[Value]: ...
+ def nodes(self) -> Iterator[Node]: ...
+ def paramNode(self) -> Node: ...
+ def returnNode(self) -> Node: ...
+ def owningNode(self) -> Node: ...
+ def registerOutput(self, n: Value) -> _int: ...
  ...
 
 # Defined in torch/csrc/jit/ir/ir.h
 class Node:
+ def __getitem__(self, key: str) -> Any: ...
  def schema(self) -> str: ...
+ def input(self) -> Value: ...
+ def inputs(self) -> List[Value]: ...
  def output(self) -> Value: ...
  def outputs(self) -> List[Value]: ...
  def outputsSize(self) -> _int: ...
  def blocks(self) -> List[Block]: ...
+ def addBlock(self) -> Block: ...
  def mustBeNone(self) -> _bool: ...
- def kindOf(self, str) -> str: ...
- def __getitem__(self, key: str) -> Any: ...
- def namedInput(self, str) -> Value: ...
- def sourceRange(self) -> SourceRange: ...
  def kind(self) -> str: ...
+ def kindOf(self, name: str) -> str: ...
+ def addInput(self, name: str) -> Value: ...
+ def replaceInput(self, i: _int, newValue: Value) -> Value: ...
+ def replaceInputWith(self, from_: Value, to: Value) -> None: ...
+ def replaceAllUsesWith(self, n: Node) -> None: ...
+ def insertBefore(self, n: Node) -> Node: ...
+ def insertAfter(self, n: Node) -> Node: ...
+ def isBefore(self, n: Node) -> _bool: ...
+ def isAfter(self, n: Node) -> _bool: ...
+ def moveBefore(self, n: Node) -> None: ...
+ def moveAfter(self, n: Node) -> None: ...
+ def removeInput(self, i: _int) -> None: ...
+ def removeAllInputs(self, i: _int) -> None: ...
+ def hasUses(self) -> _bool: ...
+ def eraseOutput(self, i: _int) -> None: ...
+ def addOutput(self) -> Value: ...
+ def scopeName(self) -> str: ...
+ def isNondeterministic(self) -> _bool: ...
+ def copyAttributes(self, rhs: Node) -> Node: ...
+ def hasAttributes(self, name: str) -> _bool: ...
+ def namedInput(self, name: str) -> Value: ...
+ def sourceRange(self) -> SourceRange: ...
  ...
 
 # Defined in torch/torch/csrc/jit/ir/ir.h
 class Graph:
+ def inputs(self) -> List[Value]: ...
+ def outputs(self) -> List[Value]: ...
+ def nodes(self) -> Iterator[Node]: ...
+ def param_node(self) -> Node: ...
+ def return_node(self) -> Node: ...
+ def addInput(self, name: str) -> Value: ...
  def eraseInput(self, i: _int) -> None: ...
+ def registerOutput(self, n: Value) -> _int: ...
+ def eraseOutput(self, i: _int) -> None: ...
+ def create(self, name: str, args, num_outputs: _int) -> Node: ...
+ def appendNode(self, n: Node) -> Node: ...
+ def prependNode(self, n: Node) -> Node: ...
+ def insertNode(self, n: Node) -> Node: ...
+ def block(self) -> Block: ...
+ def lint(self) -> None: ...
  def alias_db(self) -> AliasDb: ...
- def inputs(self) -> List[Value]: ...
  def setInsertPoint(self, n: Union[Block, Node]) -> None: ...
  def insert_point_guard(self, n: Union[Block, Node]) -> _InsertPoint: ...
  def insertPoint(self) -> Node: ...
  def insertGraph(self, callee: Graph, inputs: List[Value]) -> List[Value]: ...
  def makeMultiOutputIntoTuple(self) -> None: ...
- def nodes(self) -> Iterator: ...
  ...