Searched defs:kernel_func (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration.h | 285 …tion_type<FuncType>::value, Options&&> kernel(DispatchKey dispatch_key, FuncType* kernel_func) && { in kernel()
300 …if_t<guts::is_function_type<FuncType>::value, Options&&> catchAllKernel(FuncType* kernel_func) && { in catchAllKernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/ |
| H A D | kernel_function_test.cpp | 685 template<class Return, class... Args> struct kernel_func final { struct
686 static Return func(Args...) { return {}; } in func()
688 template<class... Args> struct kernel_func<void, Args...> final { struct
689 static void func(Args...) {} in func()
|
| H A D | kernel_function_legacy_test.cpp | 928 template<class Return, class... Args> struct kernel_func final { struct
929 static Return func(Args...) { return {}; } in func()
931 template<class... Args> struct kernel_func<void, Args...> final { struct
932 static void func(Args...) {} in func()
|
| H A D | WrapFunctionIntoRuntimeFunctor.h | 14 explicit WrapFunctionIntoRuntimeFunctor_(FuncType_&& kernel_func) in WrapFunctionIntoRuntimeFunctor_()
|
| /aosp_15_r20/external/bcc/src/cc/api/ |
| H A D | BPF.cc | 228 StatusTuple BPF::attach_kprobe(const std::string& kernel_func, in attach_kprobe()
495 StatusTuple BPF::detach_kprobe(const std::string& kernel_func, in detach_kprobe()
785 std::string BPF::get_kprobe_event(const std::string& kernel_func, in get_kprobe_event()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | kernel_ops_pattern.cc | 416 mlir::gpu::GPUFuncOp kernel_func = rewriter.create<mlir::gpu::GPUFuncOp>( in Rewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 1591 auto kernel_func = in EmitLaunchFunc() local
|