Home
last modified time | relevance | path

Searched refs:PythonSymNodeImpl (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_symnode.h33 class PythonSymNodeImpl : public c10::SymNodeImpl {
35 PythonSymNodeImpl(py::object pyobj) : c10::SymNodeImpl() { in PythonSymNodeImpl() function
43 return c10::make_intrusive<PythonSymNodeImpl>(std::move(r)); in wrap_int()
49 return c10::make_intrusive<PythonSymNodeImpl>(std::move(r)); in wrap_float()
55 return c10::make_intrusive<PythonSymNodeImpl>(std::move(r)); in wrap_bool()
64 return c10::make_intrusive<PythonSymNodeImpl>(std::move(r)); \
167 auto pother = dynamic_cast<PythonSymNodeImpl*>(other.get()); in dispatch_sym_ite_()
168 auto pthird = dynamic_cast<PythonSymNodeImpl*>(third.get()); in dispatch_sym_ite_()
173 return c10::make_intrusive<PythonSymNodeImpl>(r); in dispatch_sym_ite_()
177 auto pother = dynamic_cast<PythonSymNodeImpl*>(other.get()); in dispatch_common_()
[all …]
H A Dpybind.cpp16 c10::make_intrusive<torch::impl::PythonSymNodeImpl>(node))); in load()
45 auto* py_node = dynamic_cast<torch::impl::PythonSymNodeImpl*>( in cast()
68 c10::make_intrusive<torch::impl::PythonSymNodeImpl>(src.attr("node")))); in load()
87 dynamic_cast<torch::impl::PythonSymNodeImpl*>(si.toSymNodeImpl().get()); in cast()
98 c10::make_intrusive<torch::impl::PythonSymNodeImpl>(src.attr("node")))); in load()
119 dynamic_cast<torch::impl::PythonSymNodeImpl*>(si.toSymNodeImpl().get()); in cast()
/aosp_15_r20/external/pytorch/torch/csrc/
H A Dutils.cpp433 auto* py_node = dynamic_cast<torch::impl::PythonSymNodeImpl*>(src[i].get()); in cast()