Searched defs:OperatorHandle (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/ |
| H A D | kernel_stackbased_test.cpp | 21 void errorKernel(const OperatorHandle&, Stack* stack) { in errorKernel()
25 void incrementKernel(const OperatorHandle&, Stack* stack) { in incrementKernel()
31 void decrementKernel(const OperatorHandle&, Stack* stack) { in decrementKernel()
129 void kernelWithoutInputs(const OperatorHandle&, Stack*) { in kernelWithoutInputs()
148 void kernelWithoutTensorInputs(const OperatorHandle&, Stack* stack) { in kernelWithoutTensorInputs()
167 void kernelForSchemaInference(const OperatorHandle&, Stack* stack) { in kernelForSchemaInference()
|
| H A D | make_boxed_from_unboxed_functor.h | 16 class OperatorHandle; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/ |
| H A D | KernelFunction_test.cpp | 36 void boxed_func_with_return(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_with_return()
46 void boxed_func_without_return(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_without_return()
55 void boxed_func_with_multi_return(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_with_multi_return()
120 void boxed_func_for_inplace_op(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_for_inplace_op()
136 void boxed_func_for_outofplace_op(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_for_outofplace_op()
152 void boxed_func_for_outofplace_multi_op(const OperatorHandle& /*opHandle*/, Stack* stack) { in boxed_func_for_outofplace_multi_op()
|
| H A D | BoxedKernel.h | 12 class OperatorHandle; variable
|
| H A D | KernelFunction.h | 15 class OperatorHandle; variable
|
| H A D | KernelFunction.cpp | 13 void fallthrough_kernel(OperatorKernel*, const OperatorHandle&, DispatchKeySet, Stack*) { in fallthrough_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | Dispatcher.h | 37 class TORCH_API OperatorHandle; variable
90 friend class OperatorHandle; variable
490 explicit OperatorHandle(std::list<Dispatcher::OperatorDef>::iterator operatorIterator) in OperatorHandle() function
542 friend class OperatorHandle; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | record_function.h | 15 class TORCH_API OperatorHandle; variable
|
| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | PyInterpreter.h | 19 class OperatorHandle; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 305 void stackBasedKernel(const OperatorHandle&, c10::Stack* stack) { in stackBasedKernel()
|