pytorch/caffe2/core
Sebastian Messmer aa2862b843 Hide the OperatorKernel* argument from the stack based kernel API (#29337)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29337

This argument is needed by boxing wrappers so they're able to get a pointer to the corresponding unboxed kernel and call into it.
But if a kernel is registered in a boxed way, we don't need it and should hide this from the API.
This is especially needed for the backend fallback API where users would only be left wondering why this argument is there and what it does.
Also, hiding it allows us to potentially totally remove it in a future refactoring if we find some way to do so.
ghstack-source-id: 94481316

Test Plan: unit tests

Differential Revision: D18361991

fbshipit-source-id: 5cef26c896fe3f2a5db730d3bc79dcd62e7ef492
2019-11-23 15:25:01 -08:00
..
hip
nomnigraph Consolidate the places that find pybind11 include dirs (#29659) 2019-11-12 14:51:56 -08:00
__init__.py
allocator.cc
allocator.h
asan.h
blob.h
blob_gpu_test.cc
blob_serialization.cc
blob_serialization.h
blob_serialization_gpu.cc
blob_serializer_base.h
blob_stats.cc
blob_stats.h
blob_test.cc
CMakeLists.txt
common.cc
common.h
common_cudnn.cc
common_cudnn.h
common_gpu.cc
common_gpu.h Revert D18632773: Split libtorch.so back into libtorch_{cpu,cuda,hip} 2019-11-21 15:01:09 -08:00
common_omp.h
common_test.cc
context.cc
context.h
context_base.cc Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
context_base.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
context_gpu.cu
context_gpu.h
context_gpu_test.cc
context_test.cc
cudnn_wrappers.h
db.cc
db.h
event.cc
event.h
event_cpu.h
event_gpu.cc Add error message if CUDA startup fails (#29670) 2019-11-13 16:48:40 -08:00
event_gpu_test.cc
event_test.cc
export_c10_op_to_caffe2.cc
export_c10_op_to_caffe2.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
export_caffe2_op_to_c10.h Hide the OperatorKernel* argument from the stack based kernel API (#29337) 2019-11-23 15:25:01 -08:00
flags.h
graph.cc
graph.h
graph_test.cc
init.cc
init.h
init_denormals.cc
init_intrinsics_check.cc
init_omp.cc
init_test.cc
int8_serialization.cc
logging.h
macros.h
macros.h.in
memonger.cc
memonger.h
module.cc
module.h
module_test.cc
net.cc
net.h
net_async_base.cc Handling of failing and terminal async cpu ops (#29052) 2019-11-04 12:01:21 -08:00
net_async_base.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
net_async_scheduling.cc AsyncIf op implementation 2019-11-07 08:51:31 -08:00
net_async_scheduling.h AsyncIf op implementation 2019-11-07 08:51:31 -08:00
net_async_task.cc
net_async_task.h
net_async_task_future.cc
net_async_task_future.h
net_async_task_graph.cc
net_async_task_graph.h
net_async_tracing.cc
net_async_tracing.h
net_async_tracing_test.cc
net_dag_utils.cc
net_dag_utils.h
net_dag_utils_test.cc
net_gpu_test.cc
net_parallel.cc
net_parallel.h
net_simple.cc
net_simple.h
net_simple_refcount.cc
net_simple_refcount.h
net_simple_refcount_test.cc
net_test.cc Handling of failing and terminal async cpu ops (#29052) 2019-11-04 12:01:21 -08:00
numa.cc
numa.h
observer.h
observer_test.cc
operator.cc Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
operator.h Support Offline Tensors through ONNXIFI layer 2019-10-31 10:33:42 -07:00
operator_gpu_test.cc
operator_gradient.h
operator_schema.cc
operator_schema.h
operator_schema_test.cc
operator_test.cc
parallel_net_test.cc
plan_executor.cc
plan_executor.h
prof_dag_counters.cc
prof_dag_counters.h
qtensor.cc
qtensor.h
qtensor_serialization.cc
qtensor_serialization.h
scope_guard.h
static_tracepoint.h
static_tracepoint_elfx86.h
stats.cc
stats.h
stats_test.cc
storage.h
tensor.cc Remove TensorImpl::is_variable, deprecate Tensor::is_variable (#29653) 2019-11-14 11:41:02 -08:00
tensor.h Remove TensorImpl::is_variable, deprecate Tensor::is_variable (#29653) 2019-11-14 11:41:02 -08:00
tensor_impl.h Remove Tensor.h, TensorMethods.h from src/core. (#27086) 2019-10-06 09:37:50 -07:00
tensor_int8.cc
tensor_int8.h
test_utils.cc
test_utils.h Add randomFill to test_utils.h 2019-10-04 18:29:22 -07:00
timer.h
timer_test.cc
transform.cc
transform.h
transform_test.cc
types.cc
types.h
workspace.cc
workspace.h
workspace_test.cc