Lines Matching defs:Argument
48 Argument = Any # pyre-ignore variable
183 def create_arg(self, a: Argument) -> torch.fx.Node:
206 value: Argument,
210 x: Argument,
267 def make_tensor_meta(x: Argument) -> Optional[TensorMetadata]:
296 args: Tuple[Argument, ...],
297 kwargs: Dict[str, Argument], argument
305 args: Tuple[Argument, ...],
306 kwargs: Dict[str, Argument], argument
313 args: Tuple[Argument, ...],
314 kwargs: Dict[str, Argument], argument
359 self, target: str, args: Tuple[Argument, ...], kwargs: Dict[str, Argument] argument
366 args: Tuple[Argument, ...],
367 kwargs: Dict[str, Argument], argument
372 self, target: str, args: Tuple[Argument, ...], kwargs: Dict[str, Argument] argument
394 args: Tuple[Argument, ...],
395 kwargs: Dict[str, Argument], argument
457 def placeholder(self, name: str, arg: Argument, meta: NodeMetadata) -> ProxyValue:
466 args: Tuple[Argument, ...],
467 kwargs: Dict[str, Argument], argument
475 args: Tuple[Argument, ...],
485 inputs: List[Argument],
523 def output(self, results: List[Argument], meta: NodeMetadata) -> ProxyValue:
527 self, graph_module: fx.GraphModule, inputs: Tuple[Argument, ...]
583 def create_arg(self, a: Argument) -> torch.fx.Node:
603 args: Tuple[Argument, ...],
604 kwargs: Dict[str, Argument], argument
647 kwargs: Dict[str, Argument], argument
660 self, graph_module: fx.GraphModule, inputs: Tuple[Argument, ...]
717 args: Argument,
718 kwargs: Dict[str, Argument], argument