Home
last modified time | relevance | path

Searched refs:python_argdefs (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dmulti_kernel.py22 arg_defs, _, _, _ = kernel.args.python_argdefs()
165 _, call_args, _, arg_types = self.kernels[0].args.python_argdefs()
167 _, other_call_args, _, other_arg_types = kernel.args.python_argdefs()
196 _, call_args, precompile_args, _ = k.args.python_argdefs()
H A Dcpp_template.py49 _, call_args, _, _ = kernel.args.python_argdefs()
54 kernel.args.python_argdefs(),
H A Dtriton_combo_kernel.py669 _, _, signature, _ = self.args.python_argdefs()
843 argdefs, _, signature, _ = self.args.python_argdefs()
890 argdefs, call_args, signature, _ = self.args.python_argdefs()
1046 _, call_args, _, arg_types = self.args.python_argdefs()
H A Dtriton.py2406 argdefs, call_args, signature, _ = self.args.python_argdefs()
2603 argdefs, _, signature, _ = self.args.python_argdefs()
2814 _, call_args, _, arg_types = self.args.python_argdefs()
2843 _, call_args, arg_signatures, _ = self.args.python_argdefs()
H A Dsimd.py806 _, call_args, _, _ = self.args.python_argdefs()
867 argdefs, call_args, signature, _ = self.args.python_argdefs()
H A Dhalide.py1395 _, a, b, _ = self.args.python_argdefs()
H A Dcommon.py1339 def python_argdefs(self): member in KernelArgs
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_template.py70 _, call_args, _, _ = kernel.args.python_argdefs()
75 kernel.args.python_argdefs(),
H A Drocm_kernel.py125 _, call_args, _, arg_types = self.args.python_argdefs()
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcuda_template.py71 _, call_args, _, _ = kernel.args.python_argdefs()
76 kernel.args.python_argdefs(),
H A Dcuda_cpp_scheduling.py105 _, call_args, arg_signatures, _ = kernel.args.python_argdefs()
H A Dcuda_kernel.py160 _, call_args, _, arg_types = self.args.python_argdefs()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dselect_algorithm.py215 argdefs, _, signature, _ = self.args.python_argdefs()
250 arg_defs, *_ = self.args.python_argdefs()
303 arg_defs, *_ = self.args.python_argdefs()
550 _, call_args, _, arg_types = self.args.python_argdefs()