Home
last modified time | relevance | path

Searched refs:shared_ptr_class_ (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DMemPool.cpp9 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 DGraph.cpp19 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 Dinit.cpp29 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 Dinit.cpp19 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 Dinit.cpp16 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 Dinit.cpp174 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()