|
..
|
|
hip
|
|
nomnigraph
|
|
CMakeLists.txt
|
|
__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
|
|
common.cc
|
|
common.h
|
|
common_cudnn.cc
|
|
common_cudnn.h
|
|
common_gpu.cc
|
|
common_gpu.h
|
|
common_omp.h
|
|
common_test.cc
|
|
context.cc
|
|
context.h
|
|
context_base.cc
|
|
context_base.h
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
numa.cc
|
|
numa.h
|
|
observer.h
|
|
observer_test.cc
|
|
operator.cc
|
|
operator.h
|
|
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
|
|
plan_executor_test.cc
|
|
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
|
|
tensor.h
|
|
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
|