Searched refs:py_array (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 145 PyObject* py_array; in TF_Run_wrapper_helper() local
146 s = TF_TensorToPyArray(std::move(tf_outputs_safe[i]), &py_array); in TF_Run_wrapper_helper()
152 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in TF_Run_wrapper_helper()
268 PyObject* py_array; in RunCallableHelper() local
269 s = TensorToNdarray(output, &py_array); in RunCallableHelper()
275 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in RunCallableHelper()
418 PyObject* py_array; in TF_SessionRun_wrapper_helper() local
419 s = TF_TensorToPyArray(std::move(output_vals_safe[i]), &py_array); in TF_SessionRun_wrapper_helper()
425 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in TF_SessionRun_wrapper_helper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/ |
| H A D | ndarray_tensor.cc | 458 PyArrayObject* py_array = in TF_TensorToPyArray() local
461 Status s = CopyTF_TensorStringsToPyArray(tensor.get(), nelems, py_array); in TF_TensorToPyArray()
465 } else if (static_cast<size_t>(PyArray_NBYTES(py_array)) != in TF_TensorToPyArray()
467 return errors::Internal("ndarray was ", PyArray_NBYTES(py_array), in TF_TensorToPyArray()
471 FastMemcpy(PyArray_DATA(py_array), TF_TensorData(tensor.get()), in TF_TensorToPyArray()
472 PyArray_NBYTES(py_array)); in TF_TensorToPyArray()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | pywrap_tensor.cc | 710 auto* py_array = TFE_TensorHandleToNumpy(self->handle, &self->status); in EagerTensor_numpy_internal() local
712 Py_XDECREF(py_array); in EagerTensor_numpy_internal()
717 return PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)); in EagerTensor_numpy_internal()
847 auto py_array = tensorflow::make_safe( in EagerTensor_getbuffer() local
855 if (PyObject_GetBuffer(py_array.get(), view, flags) < 0) { in EagerTensor_getbuffer()
|
| /aosp_15_r20/external/bcc/tests/python/ |
| H A D | CMakeLists.txt | 48 add_test(NAME py_array WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
49 COMMAND ${TEST_WRAPPER} py_array sudo ${CMAKE_CURRENT_SOURCE_DIR}/test_array.py)
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| H A D | interpreter_wrapper.cc | 739 PyArrayObject* py_array = reinterpret_cast<PyArrayObject*>(py_object); in GetTensor() local
740 PyObject** data = reinterpret_cast<PyObject**>(PyArray_DATA(py_array)); in GetTensor()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 2010 py_array = np.arange(15)
2011 ret_py_obj = torch._C._ivalue_debug_python_object(py_array)
2012 self.assertEqual(py_array, ret_py_obj)
|