Searched refs:PythonMod (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/_sympy/ |
| H A D | interp.py | 35 PythonMod,
86 PythonMod: "mod", # TODO: this is wrong
|
| H A D | functions.py | 351 class PythonMod(sympy.Function): class
|
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | sym_node.py | 668 from torch.utils._sympy.functions import Mod, PythonMod
673 return PythonMod(a, b)
1077 from torch.utils._sympy.functions import Mod, PythonMod
1091 out = PythonMod(self.expr, other.expr)
|
| H A D | symbolic_shapes.py | 66 …Application, FloorDiv, Mod, PythonMod, IsNonOverlappingAndDenseIndicator, CleanDiv, FloorToInt, Ce…
1681 PythonMod,
1746 if expr.has(PythonMod):
1747 expr = expr.replace(PythonMod, mod_handler)
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | serialize.py | 1868 "PythonMod": torch.utils._sympy.functions.PythonMod,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 12196 class PythonMod(torch.nn.Module): class
12204 pm = PythonMod()
12452 class PythonMod(torch.nn.Module): class
12465 self.pm = PythonMod()
|