Home
last modified time | relevance | path

Searched refs:method_to_operator (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsym_node.py1041 def method_to_operator(method): function
1060 op = method_to_operator(method)
1138 op = method_to_operator(method)
1468 return (method_to_operator(method))(get_constant(self))
1479 return (method_to_operator(method))(get_constant(self), other)
1495 return (method_to_operator(method))(get_constant(self), other)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py36 from torch.fx.experimental.sym_node import magic_methods, method_to_operator
159 magic_ops = {method_to_operator(m) for m in magic_methods}
H A Dlowering.py35 from torch.fx.experimental.sym_node import magic_methods, method_to_operator
6104 register_lowering(method_to_operator(method))(func)
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dpartitioners.py20 from torch.fx.experimental.sym_node import magic_methods, method_to_operator
1324 default_recomputable_ops += [method_to_operator(m) for m in magic_methods]
/aosp_15_r20/external/pytorch/test/
H A Dtest_dynamic_shapes.py20 from torch.fx.experimental.sym_node import method_to_operator, SymNode, to_node
1144 lambda_apply = method_to_operator(fn)