Searched refs:py_tensor (Results 1 – 4 of 4) sorted by relevance
193 PyObject* py_tensor = THPVariable_Wrap(prepared_tensor); in tensor_to_numpy() local194 if (!py_tensor) in tensor_to_numpy()196 if (PyArray_SetBaseObject((PyArrayObject*)array.get(), py_tensor) == -1) { in tensor_to_numpy()
1776 auto py_tensor = py::cast(tensor); in tensor_slow() local1777 if (PyObject_SetAttrString(py_tensor.ptr(), "_wrapped_number", obj) < 0) { in tensor_slow()
103 auto py_tensor = py::cast(tensor); in toIValue() local105 py_tensor.ptr(), "_wrapped_number", obj.ptr()) < 0) { in toIValue()578 auto py_tensor = py::cast(tensor); in toPyObject() local579 if (PyObject_HasAttrString(py_tensor.ptr(), "_wrapped_number")) { in toPyObject()580 return py_tensor.attr("_wrapped_number"); in toPyObject()
1434 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"))