Home
last modified time | relevance | path

Searched refs:PyObjectType (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dpython_functions.cpp38 PyObjectType::get()); in toPyIValue()
48 return jit::toIValue(value, PyObjectType::get()); in toPyIValue()
139 auto child = messageJitFuture->createInstance(PyObjectType::get()); in toPyJitFuture()
348 auto userRRef = ctx.createUserRRef(dst.id_, PyObjectType::get()); in pyRemotePythonUdf()
366 auto ownerRRef = ctx.createOwnerRRef(PyObjectType::get()); in pyRemotePythonUdf()
H A Dpy_rref.cpp97 return PyObjectType::get(); in tryInferTypeWithTypeHint()
112 return PyObjectType::get(); in tryInferTypeWithTypeHint()
H A Drref_impl.h220 return type_ == PyObjectType::get(); in isPyObj()
H A Dpython_rpc_handler.cpp43 return PyObjectType::get(); in resolveType()
H A Drequest_callback_impl.cpp131 at::PyObjectType::get()); in runPythonFunction()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.h168 return toIValue(pf->func_(pyFut), PyObjectType::get()); in then()
186 PyObjectType::get())); in then()
223 IValue value = toIValue(pyValue, PyObjectType::get()); in markCompleted()
259 auto type = PyObjectType::get(); in PythonAwaitWrapper()
270 return toIValue(fg->func_(*args), PyObjectType::get()); in PythonAwaitWrapper()
273 PyObjectType::get(), std::move(f)); in PythonAwaitWrapper()
H A Dpython_ir.cpp1019 py::class_<PyObjectType, Type, PyObjectTypePtr>(m, "PyObjectType") in initPythonIRBindings()
1020 .def_static("get", &PyObjectType::get); in initPythonIRBindings()
H A Dpybind_utils.cpp526 case TypeKind::PyObjectType: { in toIValue()
H A Dinit.cpp2036 PyObjectType::get(), std::move(devices))); in initJITBindings()
H A Dscript_init.cpp2052 IValue pyobj_ivalue = toIValue(std::move(py_obj), PyObjectType::get()); in initJitScriptBindings()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Djit_type.h1674 struct PyObjectType;
1675 using PyObjectTypePtr = SingletonTypePtr<PyObjectType>;
1677 struct TORCH_API PyObjectType : public Type { struct
1684 static const TypeKind Kind = TypeKind::PyObjectType; argument
1688 PyObjectType() in PyObjectType() function
1689 : Type(TypeKind::PyObjectType) {} in PyObjectType()
H A Djit_type_base.h48 _(PyObjectType) \
106 TORCH_DECLARE_SINGLETON(PyObjectType);
H A Dtype.cpp239 PyObjectTypePtr PyObjectType::get() { in get()
240 static PyObjectTypePtr value(new PyObjectType()); in get()
H A Divalue.cpp131 return PyObjectType::get(); in get()
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dunpickler.cpp74 case PyObjectType::Kind: in restoreAccurateTypeTags()