Home
last modified time | relevance | path

Searched refs:py_obj (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_ivalue.h15 static c10::intrusive_ptr<PyObjectHolder> create(py::object py_obj) { in create()
16 return c10::make_intrusive<ConcretePyObjectHolder>(std::move(py_obj)); in create()
90 explicit ConcretePyObjectHolder(py::object py_obj) in ConcretePyObjectHolder()
91 : py_obj_(std::move(py_obj)) {} in ConcretePyObjectHolder()
H A Dpybind_utils.h124 py::object py_obj = toPyObject(fut->value()); in value() local
129 (*unwrap_func)(py_obj); in value()
131 return py_obj; in value()
H A Dscript_init.cpp2050 m.def("_ivalue_debug_python_object", [](py::object py_obj) { in initJitScriptBindings() argument
2052 IValue pyobj_ivalue = toIValue(std::move(py_obj), PyObjectType::get()); in initJitScriptBindings()
H A Dpybind_utils.cpp560 py::object py_obj = py::reinterpret_borrow<py::object>(obj); in toIValue() local
/aosp_15_r20/external/pytorch/torch/fx/passes/
H A Dgraph_drawer.py170 py_obj = module
174 if not hasattr(py_obj, atom):
176 str(py_obj) + " does not have attribute " + atom + "!"
178 py_obj = getattr(py_obj, atom)
179 return py_obj
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dpy_checkpoint_reader_wrapper.cc105 PyObject* py_obj = Py_None; in CheckpointReader_GetTensor() local
113 tensorflow::TensorToNdarray(*tensor, &py_obj)); in CheckpointReader_GetTensor()
116 PyArray_Return(reinterpret_cast<PyArrayObject*>(py_obj))); in CheckpointReader_GetTensor()
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/upb/python/
Dprotobuf.c144 void PyUpb_WeakMap_Add(PyUpb_WeakMap* map, const void* key, PyObject* py_obj) { in PyUpb_WeakMap_Add() argument
146 upb_value_ptr(py_obj), map->arena); in PyUpb_WeakMap_Add()
195 void PyUpb_ObjCache_Add(const void* key, PyObject* py_obj) { in PyUpb_ObjCache_Add() argument
196 PyUpb_WeakMap_Add(PyUpb_ObjCache_Instance(), key, py_obj); in PyUpb_ObjCache_Add()
Dprotobuf.h149 void PyUpb_WeakMap_Add(PyUpb_WeakMap* map, const void* key, PyObject* py_obj);
178 void PyUpb_ObjCache_Add(const void* key, PyObject* py_obj);
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Divalue.cpp193 c10::intrusive_ptr<at::ivalue::PyObjectHolder> py_obj = toPyObjectHolder(); in visit() local
194 auto match = py_obj->tryToInferType(); in visit()
196 auto contained_value = py_obj->toIValue(match.type()); in visit()
246 c10::intrusive_ptr<at::ivalue::PyObjectHolder> py_obj = toPyObjectHolder(); in getSubValues() local
247 auto match = py_obj->tryToInferType(); in getSubValues()
249 "Cannot infer type of ", py_obj->toStr(), ": ", match.reason()); in getSubValues()
250 auto contained_value = py_obj->toIValue(match.type()); in getSubValues()
850 auto py_obj = v.toPyObject(); in operator <<() local
851 return out << "<PyObject at" << py_obj << ">"; in operator <<()
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtrace_rules.py3420 py_obj: Optional[object]
3465 py_obj = obj.get_function()
3467 py_obj = None
3468 fi = FunctionInfo(py_obj, obj.get_name(), obj.get_filename(), obj.get_code())
3480 rule = lookup_inner(fi.py_obj, fi.name, fi.filename, is_inlined_call, reasons)
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp2467 PyObject* py_obj = unflatten(outputs, desc); in ONNXAssignOutputShape() local
2468 TORCH_INTERNAL_ASSERT(PyTuple_Check(py_obj)); in ONNXAssignOutputShape()
2473 py_obj, in ONNXAssignOutputShape()
2482 Py_DECREF(py_obj); in ONNXAssignOutputShape()