Searched refs:shared_ptr_class_ (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | MemPool.cpp | 9 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
13 shared_ptr_class_<::c10::cuda::MemPool>(torch_C_m, "_MemPool") in THCPMemPool_init()
17 shared_ptr_class_<::c10::cuda::MemPoolContext>(torch_C_m, "_MemPoolContext") in THCPMemPool_init()
|
| H A D | Graph.cpp | 19 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
28 shared_ptr_class_<::at::cuda::CUDAGraph>(torch_C_m, "_CUDAGraph") in THCPGraph_init()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/ |
| H A D | init.cpp | 29 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
50 shared_ptr_class_<RpcBackendOptions>( in rpc_init()
82 shared_ptr_class_<WorkerInfo>( in rpc_init()
133 shared_ptr_class_<RpcAgent>(module, "RpcAgent") in rpc_init()
174 shared_ptr_class_<PyRRef>(module, "PyRRef", R"( in rpc_init()
572 shared_ptr_class_<TensorPipeAgent>(module, "TensorPipeAgent", rpcAgent) in rpc_init()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/testing/ |
| H A D | init.cpp | 19 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
38 shared_ptr_class_<FaultyTensorPipeRpcBackendOptions>( in faulty_agent_init()
67 shared_ptr_class_<FaultyTensorPipeAgent>( in faulty_agent_init()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/ |
| H A D | init.cpp | 16 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
37 shared_ptr_class_<DistAutogradContext>(module, "DistAutogradContext") in dist_autograd_init()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | init.cpp | 174 using shared_ptr_class_ = py::class_<T, std::shared_ptr<T>>; typedef
466 shared_ptr_class_<::c10d::GradBucket>( in c10d_init()
538 shared_ptr_class_<::c10d::Reducer>(module, "Reducer") in c10d_init()
695 shared_ptr_class_<::c10d::Logger>(module, "Logger") in c10d_init()
2563 shared_ptr_class_<::gloo::transport::Device>(processGroupGloo, "Device"); in c10d_init()
|