Home
last modified time | relevance | path

Searched refs:cpp_kernel_key (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dmkldnn_ir.py263 self.cpp_kernel_key,
338 self.cpp_kernel_key,
433 self.cpp_kernel_key,
522 self.cpp_kernel_key,
721 self.cpp_kernel_key,
986 self.cpp_kernel_key,
1122 self.cpp_kernel_key,
1163 self.cpp_kernel_key = "linear_pointwise"
1180 self.cpp_kernel_key,
1244 self.cpp_kernel_key,
[all …]
H A Dir.py5417 …self.cpp_kernel_key = f"{self.cpp_kernel_name.replace('::', '_')}_{self.cpp_kernel_overload_name}"…
5602 self.cpp_kernel_key,
6000 self.cpp_kernel_key = "convolution_pointwise"
6021 self.cpp_kernel_key,
6072 self.cpp_kernel_key = "convolution_pointwise_binary"
6097 self.cpp_kernel_key,
6163 self.cpp_kernel_key = "convolution_pointwise_binary_"
6188 self.cpp_kernel_key,
6259 self.cpp_kernel_key = "mkl_linear"
6275 self.cpp_kernel_key,
[all …]
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_wrapper_cpu.py2124 cpp_kernel_key: str,
2157 cpp_kernel_key,
2169 cpp_kernel_key,
2298 cpp_kernel_key: str,
2306 if cpp_kernel_key not in self.extern_call_ops:
2314 self.extern_call_ops.add(cpp_kernel_key)
2375 cpp_kernel_key, argument
2401 self.extern_call_ops.add(cpp_kernel_key)
H A Dwrapper.py847 cpp_kernel_key: str,