Searched refs:py_buffer (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 456 auto py_buffer = absl::make_unique<PjRtStreamExecutorBuffer>( in AllocateDestinationBuffer() local
468 RecordUsage(py_buffer->GetBufferWithUsageHold(), local_device, local_device, in AllocateDestinationBuffer()
473 return py_buffer; in AllocateDestinationBuffer()
721 std::unique_ptr<PjRtStreamExecutorBuffer> py_buffer, in BufferFromHostBuffer()
727 py_buffer->GetBufferWithUsageHold()); in BufferFromHostBuffer()
765 py_buffer{py_buffer.get()}, in BufferFromHostBuffer()
766 on_device_shape{py_buffer->on_device_shape()}, in BufferFromHostBuffer()
824 return std::unique_ptr<PjRtBuffer>(std::move(py_buffer)); in BufferFromHostBuffer()
850 std::unique_ptr<PjRtStreamExecutorBuffer> py_buffer, in CreateUninitializedBuffer()
855 return std::unique_ptr<PjRtBuffer>(std::move(py_buffer)); in CreateUninitializedBuffer()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | dlpack.cc | 250 StatusOr<py::capsule> BufferToDLPackManagedTensor(py::handle py_buffer, in BufferToDLPackManagedTensor() argument
252 PyBuffer* buffer = py::cast<PyBuffer*>(py_buffer); in BufferToDLPackManagedTensor()
281 pack->buffer_reference = py::reinterpret_borrow<py::object>(py_buffer); in BufferToDLPackManagedTensor()
|
| D | BUILD | 183 "py_buffer.cc",
188 "py_buffer.h",
|
| /external/llvm-project/lldb/source/Plugins/ScriptInterpreter/Python/ |
| D | PythonDataObjects.cpp | 1219 Py_buffer py_buffer = {}; in Create() local
1220 PyObject_GetBuffer(obj.get(), &py_buffer, flags); in Create()
1221 if (!py_buffer.obj) in Create()
1223 return PythonBuffer(py_buffer); in Create()
1240 PythonBuffer(const Py_buffer &py_buffer) : m_buffer(py_buffer) {} in PythonBuffer() argument
|