| /external/executorch/devtools/debug_format/ |
| D | et_schema.py | 219 for fx_node in graph.nodes:
221 fx_node.target == exir.memory.alloc
222 or fx_node.target == operator.getitem
225 op = fx_node.op
226 name = FXOperatorGraph._get_node_name(fx_node)
227 target = fx_node.target
228 args = fx_node.args
229 kwargs = fx_node.kwargs
230 metadata = FXOperatorGraph._extract_metadata(fx_node.meta, skip_stack_trace)
232 fx_node.meta.get("val")
[all …]
|
| /external/pytorch/torch/fx/experimental/ |
| D | sym_node.py | 75 fx_node=None, argument
146 self.fx_node = tx_validation_en and fx_node
150 self._expr, shape_env, self.pytype, self._hint, self.constant, self.fx_node
160 and self.fx_node == other.fx_node
165 return hash((self._expr, self.pytype, self._hint, self.constant, self.fx_node))
233 sympy.Integer(num), self.shape_env, int, num, constant=num, fx_node=num
241 sympy.Float(num), self.shape_env, float, num, constant=num, fx_node=num
254 fx_node=num,
274 if self.fx_node is not None:
429 r = self.shape_env.evaluate_expr(self.expr, self.hint, fx_node=self.fx_node)
[all …]
|
| D | symbolic_shapes.py | 3235 fx_node = self._create_fx_placeholder_and_z3var(symbol, int)
3240 fx_node = None
3254 out = SymInt(SymNode(sym, self, int, hint, fx_node=fx_node))
3274 fx_node = self._create_fx_placeholder_and_z3var(symbol, float)
3279 fx_node = None
3292 out = SymFloat(SymNode(sym, self, float, hint, fx_node=fx_node))
3335 fx_node = self._create_fx_placeholder_and_z3var(symbol, float)
3339 return SymFloat(SymNode(symbol, self, float, None, fx_node=fx_node))
3354 fx_node = self._create_fx_placeholder_and_z3var(symbol, int)
3358 return SymInt(SymNode(symbol, self, int, None, fx_node=fx_node))
[all …]
|
| /external/pytorch/torch/_export/ |
| D | converter.py | 545 fx_node = self.name_to_node[name]
547 fx_node = self.name_to_constant[name]
549 fx_node = self.name_to_non_tensor_attribute_node[name]
551 fx_node = self.name_to_non_tensor_attribute[name]
554 return fx_node
595 fx_node = get_node_as_placeholder_or_get_attr(
608 fx_node = get_node_as_placeholder_or_get_attr(
626 fx_node = get_node_as_placeholder_or_get_attr(
641 fx_node = self.fx_graph.placeholder(normalized_name)
643 self.name_to_node[name] = fx_node
[all …]
|
| /external/executorch/exir/serde/ |
| D | serialize.py | 384 fx_node = self.graph.create_node(
388 self.deserialize_arbitrary_outputs(serialized_node, fx_node)
390 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
407 fx_node = self.graph.create_node("call_function", target, args, {}, name)
409 self.deserialize_arbitrary_outputs(serialized_node, fx_node)
411 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
423 fx_node = self.graph.create_node(
426 self.deserialize_outputs(serialized_node, fx_node)
427 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
445 fx_node = self.graph.create_node("call_function", target, args, None, None)
[all …]
|
| D | export_serialize.py | 1665 fx_node = self.graph.create_node("call_function", target, args, {}, name)
1666 self.deserialize_sym_op_outputs(serialized_node, fx_node)
1681 fx_node = self.graph.create_node(
1684 self.deserialize_outputs(serialized_node, fx_node)
1685 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
1697 fx_node = self.graph.create_node(
1700 self.deserialize_outputs(serialized_node, fx_node)
1706 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
1708 fx_node.op not in ["placeholder", "output"]
1709 and "nn_module_stack" not in fx_node.meta
[all …]
|
| /external/pytorch/torch/_subclasses/ |
| D | _fake_tensor_utils.py | 33 fx_node: torch.fx.Node
38 node._expr, node.pytype, node._hint, node.constant, node.fx_node
43 self._expr, shape_env, self.pytype, self._hint, self.constant, self.fx_node
66 and self.fx_node == other.fx_node
73 return hash((self._expr, self.pytype, self._hint, self.constant, self.fx_node))
|
| /external/pytorch/torch/onnx/_internal/fx/passes/ |
| D | modularization.py | 537 yield node.fx_node
615 fx_node = ir_node.fx_node
616 copy_env[fx_node] = fx_graph.node_copy(
617 fx_node, arg_transform=_arg_transform
712 def fx_node(self) -> torch.fx.Node: member in _LeafNode
724 return self.fx_node.meta.get("stack_trace")
850 for fx_node in self.module.graph.nodes:
852 _LeafNode(fx_node, is_exported_program=self.is_exported_program)
|
| /external/pytorch/torch/_export/passes/ |
| D | replace_quantized_ops_with_standard_ops_pass.py | 261 gm: torch.fx.GraphModule, opname: str, fx_node: torch.fx.Node
264 fx_node = gm.graph.call_function(torch.ops.aten.relu, (fx_node,))
265 return fx_node
579 fx_node = fx_transform_quantized_op_to_standard_op(gm, node)
580 node.replace_all_uses_with(fx_node)
581 last_quantized_node = fx_node
584 fx_node = _transform_prepacked_op(gm, node)
585 node.replace_all_uses_with(fx_node)
586 last_quantized_node = fx_node
|
| /external/pytorch/torch/_inductor/ |
| D | debug.py | 148 fx_node = graph.call_function(node_func, args=(), kwargs=kwargs) # type: ignore[arg-type]
160 outputs.append(fx_node)
162 fx_node.name = name
164 fx_node.meta["fusion_meta"] = FusionMeta(group, snode, node_type)
166 node_to_fx_node[name] = fx_node
168 buf_to_fx_node[buf.get_name()] = fx_node
171 first_node = fx_node
178 fx_node = node_to_fx_node[name]
187 if dep_node == fx_node: # to avoid cycles
191 fx_node.args = tuple(new_args)
|
| D | lowering.py | 2110 def constrain_to_fx_strides(fx_node, *args, ignore_mutated_args_FIXME=False, **kwargs): argument
2123 assert isinstance(fx_node.target, torch._ops.OpOverload)
2124 schema = fx_node.target._schema
2134 for idx, (arg, fx_arg) in enumerate(zip(args, fx_node.args)):
2142 fx_arg = fx_node.kwargs[key]
2149 apply_constraint(arg, fx_arg) for arg, fx_arg in zip(args, fx_node.args)
2151 kwargs = {k: apply_constraint(v, fx_node.kwargs[k]) for k, v in kwargs.items()}
2162 def sdpa_constraint(fx_node, *args, **kwargs): argument
2219 apply_constraint(arg, fx_arg) for arg, fx_arg in zip(args, fx_node.args)
2221 kwargs = {k: apply_constraint(v, fx_node.kwargs[k]) for k, v in kwargs.items()}
|
| D | graph.py | 1061 for r, fx_node in zip(result, fx_node_args):
1069 r, fx_node.meta["val"].stride()
|
| D | utils.py | 651 and (fx_node := getattr(V.interpreter, "current_node", None))
652 and fx_node.target != "index_expr"
|
| D | scheduler.py | 668 self.node.fx_node
|
| D | ir.py | 3992 self.fx_node = V.graph.current_node
|
| /external/pytorch/torch/_export/serde/ |
| D | serialize.py | 1693 fx_node = self.graph.create_node("call_function", target, args, {}, name)
1694 self.deserialize_sym_op_outputs(serialized_node, fx_node)
1709 fx_node = self.graph.create_node(
1712 self.deserialize_outputs(serialized_node, fx_node)
1713 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
1725 fx_node = self.graph.create_node(
1728 self.deserialize_outputs(serialized_node, fx_node)
1734 fx_node.meta.update(self.deserialize_metadata(serialized_node.metadata))
1735 if fx_node.op not in ["placeholder", "output"] and "nn_module_stack" not in fx_node.meta:
1736 fx_node.meta["nn_module_stack"] = {} # serialization throws away empty dicts
[all …]
|
| /external/pytorch/torch/_inductor/kernel/ |
| D | conv.py | 671 def constrain_conv_to_fx_strides(fx_node, *args, **kwargs): argument
672 assert fx_node.target == torch.ops.aten.convolution.default
676 return constrain_to_fx_strides(fx_node, *args, **kwargs)
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _core.py | 279 def _set_node_metadata(fx_node: torch.fx.Node, ir_node: ir.Node) -> None:
281 namespace, class_hierarchy, name_scopes = _get_node_namespace(fx_node)
285 ir_node.metadata_props["pkg.torch.onnx.fx_node"] = str(fx_node.format_node())
286 ir_node.metadata_props["pkg.torch.onnx.stack_trace"] = fx_node.meta.get(
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | common.py | 1818 fx_node = V.interpreter.current_node
1819 if fx_node.target == name and self.node_to_bounds is not None:
1821 return self.node_to_bounds.get(fx_node, ValueRanges.unknown())
1826 s in fx_node.target
|
| D | cpp.py | 3637 for fx_node in sub_block.graph.nodes:
3638 if fx_node.target in ["load", "store"]:
3639 assert fx_node.meta
3640 assert OptimizationContext.key in fx_node.meta
3641 opt_ctx: OptimizationContext = fx_node.meta[
|