| .. |
|
hip
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
nomnigraph
|
Update CMakeLists.txt (#87030)
|
2022-10-28 04:56:40 +00:00 |
|
__init__.py
|
|
|
|
allocator.cc
|
|
|
|
allocator.h
|
|
|
|
blob.h
|
|
|
|
blob_gpu_test.cc
|
|
|
|
blob_serialization.cc
|
[caffe2] Don't copy Tensor dims during deserialization (#79471)
|
2022-07-12 21:36:26 +00:00 |
|
blob_serialization.h
|
[caffe2] Don't copy Tensor dims during deserialization (#79471)
|
2022-07-12 21:36:26 +00:00 |
|
blob_serialization_gpu.cc
|
|
|
|
blob_serializer_base.h
|
|
|
|
blob_stats.cc
|
|
|
|
blob_stats.h
|
|
|
|
blob_test.cc
|
|
|
|
CMakeLists.txt
|
Remove caffe2 mobile (#84338)
|
2022-09-08 01:49:55 +00:00 |
|
common.cc
|
|
|
|
common.h
|
|
|
|
common_cudnn.cc
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
common_cudnn.h
|
|
|
|
common_gpu.cc
|
|
|
|
common_gpu.h
|
[CUDA] Drop CUDA 10 support (#89582)
|
2023-01-05 05:11:53 +00:00 |
|
common_omp.h
|
|
|
|
common_test.cc
|
|
|
|
context.cc
|
|
|
|
context.h
|
|
|
|
context_base.cc
|
|
|
|
context_base.h
|
|
|
|
context_gpu.cu
|
Fix C2 Ambiguous namespace (#89534)
|
2022-12-07 23:36:41 +00:00 |
|
context_gpu.h
|
Fixes to DSA infra (#91835)
|
2023-01-12 21:54:26 +00:00 |
|
context_gpu_test.cc
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
context_test.cc
|
cleanup unused include (#93359)
|
2023-02-04 02:15:50 +00:00 |
|
cudnn_wrappers.h
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
db.cc
|
|
|
|
db.h
|
|
|
|
distributions_stubs.h
|
|
|
|
event.cc
|
|
|
|
event.h
|
|
|
|
event_cpu.h
|
|
|
|
event_gpu.cc
|
|
|
|
event_gpu_test.cc
|
|
|
|
event_test.cc
|
|
|
|
export_c10_op_to_caffe2.cc
|
|
|
|
export_c10_op_to_caffe2.h
|
|
|
|
export_caffe2_op_to_c10.h
|
[caffe2] Remove OperatorBase::newstyle_outputs_ (#67093)
|
2023-01-23 22:41:59 +00: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
|
Introduce TORCH_DISABLE_GPU_ASSERTS (#84190)
|
2022-11-04 04:43:05 +00:00 |
|
memonger.cc
|
|
|
|
memonger.h
|
|
|
|
module.cc
|
|
|
|
module.h
|
|
|
|
module_test.cc
|
|
|
|
net.cc
|
|
|
|
net.h
|
|
|
|
net_async_base.cc
|
|
|
|
net_async_base.h
|
|
|
|
net_async_scheduling.cc
|
|
|
|
net_async_scheduling.h
|
|
|
|
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
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
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
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
numa.cc
|
|
|
|
numa.h
|
|
|
|
observer.h
|
|
|
|
observer_test.cc
|
|
|
|
operator.cc
|
[caffe2] Remove OperatorBase::newstyle_outputs_ (#67093)
|
2023-01-23 22:41:59 +00:00 |
|
operator.h
|
[caffe2] Remove OperatorBase::newstyle_outputs_ (#67093)
|
2023-01-23 22:41:59 +00:00 |
|
operator_gpu_test.cc
|
|
|
|
operator_gradient.h
|
|
|
|
operator_schema.cc
|
|
|
|
operator_schema.h
|
Revert "Fix crash on unload torch cpu dll (#67632)"
|
2022-08-02 00:56:18 +00:00 |
|
operator_schema_test.cc
|
|
|
|
operator_test.cc
|
|
|
|
parallel_net_test.cc
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
plan_executor.cc
|
|
|
|
plan_executor.h
|
|
|
|
plan_executor_test.cc
|
|
|
|
prof_dag_counters.cc
|
turn on -Werror=type-limits in our Bazel CPU build
|
2022-06-10 10:04:08 +00:00 |
|
prof_dag_counters.h
|
|
|
|
qtensor.cc
|
|
|
|
qtensor.h
|
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
|
2022-07-26 01:20:44 +00:00 |
|
qtensor_serialization.cc
|
|
|
|
qtensor_serialization.h
|
|
|
|
scope_guard.h
|
|
|
|
serialization_test.cc
|
|
|
|
static_tracepoint.h
|
|
|
|
static_tracepoint_elfx86.h
|
|
|
|
stats.cc
|
|
|
|
stats.h
|
|
|
|
stats_test.cc
|
|
|
|
storage.h
|
|
|
|
tensor.cc
|
[PyTorch][caffe2] Add CAFFE2_{DECLARE,DEFINE}_KNOWN_TYPE (#83707)
|
2022-08-30 23:09:49 +00:00 |
|
tensor.h
|
[PyTorch][caffe2] Add CAFFE2_{DECLARE,DEFINE}_KNOWN_TYPE (#83707)
|
2022-08-30 23:09:49 +00:00 |
|
tensor_impl.h
|
|
|
|
tensor_int8.cc
|
|
|
|
tensor_int8.h
|
|
|
|
test_utils.cc
|
|
|
|
test_utils.h
|
|
|
|
timer.h
|
|
|
|
timer_test.cc
|
|
|
|
transform.cc
|
|
|
|
transform.h
|
|
|
|
transform_test.cc
|
|
|
|
types.cc
|
|
|
|
types.h
|
|
|
|
workspace.cc
|
|
|
|
workspace.h
|
|
|
|
workspace_test.cc
|
|
|
