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()
|