Home
last modified time | relevance | path

Searched refs:py_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_numpy.cpp193 PyObject* py_tensor = THPVariable_Wrap(prepared_tensor); in tensor_to_numpy() local
194 if (!py_tensor) in tensor_to_numpy()
196 if (PyArray_SetBaseObject((PyArrayObject*)array.get(), py_tensor) == -1) { in tensor_to_numpy()
H A Dpython_arg_parser.cpp1776 auto py_tensor = py::cast(tensor); in tensor_slow() local
1777 if (PyObject_SetAttrString(py_tensor.ptr(), "_wrapped_number", obj) < 0) { in tensor_slow()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.cpp103 auto py_tensor = py::cast(tensor); in toIValue() local
105 py_tensor.ptr(), "_wrapped_number", obj.ptr()) < 0) { in toIValue()
578 auto py_tensor = py::cast(tensor); in toPyObject() local
579 if (PyObject_HasAttrString(py_tensor.ptr(), "_wrapped_number")) { in toPyObject()
580 return py_tensor.attr("_wrapped_number"); in toPyObject()
/aosp_15_r20/external/pytorch/test/
H A Dtest_mps.py1434 py_tensor = torch.from_numpy(np_features).to(device)
1435 py_relu = torch.nn.ReLU(inplace=False)(py_tensor)
1444 py_tensor = torch.from_numpy(np_features).to(device)
1445 py_relu = torch.nn.ReLU(inplace=True)(py_tensor)
1450 self.assertEqual(np_relu, py_tensor.to("cpu"))