Home
last modified time | relevance | path

Searched refs:py_obj_ (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/python/
Dpython_ivalue.h26 return py_obj_.ptr(); in getPyObject()
31 return torch::jit::tryToInferType(py_obj_); in tryToInferType()
37 return torch::jit::toIValue(py_obj_, type, N);
42 return py::str(py_obj_); in toStr()
54 return extractorFn(py_obj_).cast<std::vector<at::Tensor>>(); in extractTensors()
82 py_obj_.dec_ref(); in ~ConcretePyObjectHolder()
85 py_obj_.ptr() = nullptr; in ~ConcretePyObjectHolder()
91 : py_obj_(std::move(py_obj)) {} in ConcretePyObjectHolder()
94 py::object py_obj_;
/external/pytorch/torch/csrc/distributed/c10d/
DPyProcessGroup.hpp52 py_obj_ = py::cast(this); in ref_py_object()
56 py::object py_obj_; member in c10d::PyProcessGroup::PyWork