Home
last modified time | relevance | path

Searched refs:getCustomClassType (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/csrc/distributed/rpc/
Drequest_callback_no_python.cpp106 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processMessage()
115 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processMessage()
150 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processScriptCall()
200 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in assignOwnerRRef()
244 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processScriptRRefFetchCall()
348 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processForwardAutogradReq()
385 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processBackwardAutogradReq()
480 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processRunWithProfilingReq()
608 at::getCustomClassType<c10::intrusive_ptr<Message>>(), in asFuture()
Drequest_callback_impl.cpp177 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processScriptCall()
192 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processPythonCall()
238 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processPythonRRefFetchCall()
296 c10::getCustomClassType<c10::intrusive_ptr<Message>>()); in processRRefBackward()
/external/pytorch/aten/src/ATen/core/
Dcustom_class.h18 const c10::ClassTypePtr& getCustomClassType() { in getCustomClassType() function
Divalue_inl.h1787 const auto* expected_type = c10::getCustomClassType<c10::intrusive_ptr<T>>().get();
1805 const auto* expected_type = c10::getCustomClassType<c10::intrusive_ptr<T>>().get();
2315 return c10::getCustomClassType<c10::intrusive_ptr<T>>();
Djit_type.h1803 return getCustomClassType<T>(); in call()
/external/pytorch/torch/csrc/jit/runtime/
Dregister_distributed_ops.cpp48 c10::getCustomClassType<c10::intrusive_ptr<dist_rpc::WorkerInfo>>() == in prepare_and_call_rpc_op()
/external/pytorch/torch/csrc/jit/python/
Dpybind_utils.h752 const auto& type = c10::getCustomClassType<c10::intrusive_ptr<T>>(); in toCustomClass()