Searched refs:py_array (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 144 PyObject* py_array; in TF_Run_wrapper_helper() local
145 s = TF_TensorToPyArray(std::move(tf_outputs_safe[i]), &py_array); in TF_Run_wrapper_helper()
151 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in TF_Run_wrapper_helper()
267 PyObject* py_array; in RunCallableHelper() local
268 s = TensorToNdarray(output, &py_array); in RunCallableHelper()
274 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in RunCallableHelper()
417 PyObject* py_array; in TF_SessionRun_wrapper_helper() local
418 s = TF_TensorToPyArray(std::move(output_vals_safe[i]), &py_array); in TF_SessionRun_wrapper_helper()
424 make_safe(PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)))); in TF_SessionRun_wrapper_helper()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 451 PyArrayObject* py_array = in TF_TensorToPyArray() local
454 Status s = CopyTF_TensorStringsToPyArray(tensor.get(), nelems, py_array); in TF_TensorToPyArray()
458 } else if (static_cast<size_t>(PyArray_NBYTES(py_array)) != in TF_TensorToPyArray()
460 return errors::Internal("ndarray was ", PyArray_NBYTES(py_array), in TF_TensorToPyArray()
464 FastMemcpy(PyArray_DATA(py_array), TF_TensorData(tensor.get()), in TF_TensorToPyArray()
465 PyArray_NBYTES(py_array)); in TF_TensorToPyArray()
|
| /external/bcc/tests/python/ |
| D | CMakeLists.txt | 52 add_test(NAME py_array WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
53 COMMAND ${TEST_WRAPPER} py_array sudo ${CMAKE_CURRENT_SOURCE_DIR}/test_array.py)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 638 auto* py_array = TFE_TensorHandleToNumpy(self->handle, &self->status); in EagerTensor_numpy_internal() local
640 Py_XDECREF(py_array); in EagerTensor_numpy_internal()
645 return PyArray_Return(reinterpret_cast<PyArrayObject*>(py_array)); in EagerTensor_numpy_internal()
742 auto py_array = tensorflow::make_safe( in EagerTensor_getbuffer() local
749 if (PyObject_GetBuffer(py_array.get(), view, flags) < 0) { in EagerTensor_getbuffer()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 662 PyArrayObject* py_array = reinterpret_cast<PyArrayObject*>(py_object); in GetTensor() local
663 PyObject** data = reinterpret_cast<PyObject**>(PyArray_DATA(py_array)); in GetTensor()
|