Home
last modified time | relevance | path

Searched defs:fx_node (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dconv.py671 def constrain_conv_to_fx_strides(fx_node, *args, **kwargs): argument
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/passes/
H A Dmodularization.py712 def fx_node(self) -> torch.fx.Node: member in _LeafNode
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsymbolic_shapes.py5119 def evaluate_expr(self, orig_expr: "sympy.Expr", hint=None, fx_node=None, argument
5130 def _evaluate_expr(self, orig_expr: "sympy.Expr", hint=None, fx_node=None, argument
5342 def defer_runtime_assert(self, orig_expr: "sympy.Expr", msg, fx_node=None): argument
H A Dsym_node.py75 fx_node=None, argument
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2110 def constrain_to_fx_strides(fx_node, *args, ignore_mutated_args_FIXME=False, **kwargs): argument
2162 def sdpa_constraint(fx_node, *args, **kwargs): argument