Home
last modified time | relevance | path

Searched refs:torch_opinfo_variant_name (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Drefs.py104 torch_opinfo_variant_name="", # the variant name for corresponding torch opinfo argument
109 self.torch_opinfo_variant_name = torch_opinfo_variant_name
111 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
133 torch_opinfo_variant_name="", # the variant name for corresponding torch opinfo argument
137 self.torch_opinfo_variant_name = torch_opinfo_variant_name
139 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
164 torch_opinfo_variant_name="", # the variant name for corresponding torch opinfo argument
169 self.torch_opinfo_variant_name = torch_opinfo_variant_name
171 torch_opinfo_name, torch_opinfo_variant_name, op_db=op_db
194 torch_opinfo_variant_name="", # the variant name for corresponding torch opinfo argument
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_methods_invocations.py21657 torch_opinfo_variant_name="number_mean",
21687 torch_opinfo_variant_name="in_place",
21755 torch_opinfo_variant_name="tensor_overload",
21805 torch_opinfo_variant_name="tensor_overload",
21826 torch_opinfo_variant_name="variadic_tensors",
21876 torch_opinfo_variant_name="list_of_tensors",
22196 torch_opinfo_variant_name="mvlgamma_p_1",
22208 torch_opinfo_variant_name="mvlgamma_p_3",
22214 torch_opinfo_variant_name="mvlgamma_p_5",
22262 torch_opinfo_variant_name="with_dtype",
[all …]