Searched refs:torch_opinfo_name (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | refs.py | 103 torch_opinfo_name, # the string name of the corresponding torch opinfo argument
108 self.torch_opinfo_name = torch_opinfo_name
111 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
132 torch_opinfo_name, # the string name of the corresponding torch opinfo argument
136 self.torch_opinfo_name = torch_opinfo_name
139 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
163 torch_opinfo_name, # the string name of the corresponding torch opinfo argument
168 self.torch_opinfo_name = torch_opinfo_name
171 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
193 torch_opinfo_name, # the string name of the corresponding torch opinfo argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | fft.py | 54 torch_opinfo_name, # the string name of the corresponding torch opinfo argument
58 self.torch_opinfo_name = torch_opinfo_name
60 torch_opinfo_name, torch_opinfo_variant, op_db=op_db
640 torch_opinfo_name="fft.fft",
644 torch_opinfo_name="fft.ifft",
648 torch_opinfo_name="fft.rfft",
652 torch_opinfo_name="fft.irfft",
656 torch_opinfo_name="fft.hfft",
660 torch_opinfo_name="fft.ihfft",
664 torch_opinfo_name="fft.fftn",
[all …]
|
| H A D | special.py | 718 torch_opinfo_name="special.bessel_j0",
731 torch_opinfo_name="special.bessel_j1",
744 torch_opinfo_name="special.entr",
758 torch_opinfo_name="special.erfcx",
770 torch_opinfo_name="special.i0e",
776 torch_opinfo_name="special.i1",
799 torch_opinfo_name="special.i1e",
804 torch_opinfo_name="special.log_ndtr",
809 torch_opinfo_name="special.ndtr",
814 torch_opinfo_name="special.ndtri",
[all …]
|
| H A D | linalg.py | 2417 torch_opinfo_name="linalg.cross",
2429 torch_opinfo_name="linalg.diagonal",
2435 torch_opinfo_name="linalg.vecdot",
2440 torch_opinfo_name="linalg.vector_norm",
2453 torch_opinfo_name="linalg.matrix_norm",
2462 torch_opinfo_name="linalg.norm",
2471 torch_opinfo_name="linalg.svd",
2477 torch_opinfo_name="linalg.svdvals",
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 21364 torch_opinfo_name="abs",
21374 torch_opinfo_name="acos",
21404 torch_opinfo_name="acosh",
21437 torch_opinfo_name="asin",
21463 torch_opinfo_name="asinh",
21490 torch_opinfo_name="lerp",
21494 torch_opinfo_name="ones",
21504 torch_opinfo_name="zeros",
21514 torch_opinfo_name="cauchy",
21537 torch_opinfo_name="exponential",
[all …]
|