Searched refs:cpp_signature (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration.h | 402 …ispatch_key, KernelFunction&& func, std::optional<impl::CppSignature> cpp_signature, std::unique_p… in kernel() argument
406 config.cpp_signature = cpp_signature; in kernel()
424 , cpp_signature(std::nullopt) in KernelRegistrationConfig()
430 std::optional<impl::CppSignature> cpp_signature; member
|
| H A D | op_registration.cpp | 135 kernel.cpp_signature, in registerOp_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | OperatorEntry.cpp | 108 std::optional<CppSignature> cpp_signature, in registerKernel() argument
119 if (cpp_signature.has_value()) { in registerKernel()
122 TORCH_CHECK(*cpp_signature == local_cpp_signature->signature, in registerKernel()
129 " kernel 2: ", cpp_signature->name(), "\n", in registerKernel()
134 local_cpp_signature = CppSignatureWithDebug { *cpp_signature, debug, dispatch_key }; in registerKernel()
|
| H A D | Dispatcher.cpp | 343 std::optional<impl::CppSignature> cpp_signature, in registerImpl() argument
355 std::move(cpp_signature), in registerImpl()
|
| H A D | OperatorEntry.h | 134 std::optional<CppSignature> cpp_signature,
|
| H A D | Dispatcher.h | 224 …ispatch_key, KernelFunction kernel, std::optional<impl::CppSignature> cpp_signature, std::unique_p…
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | library.cpp | 47 …tion(c10::KernelFunction func, std::optional<c10::impl::CppSignature> cpp_signature, std::unique_p… in CppFunction() argument
49 , cpp_signature_(cpp_signature) in CppFunction()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | library.h | 319 std::optional<c10::impl::CppSignature> cpp_signature,
|