Searched defs:py_buffer (Results 1 – 3 of 3) sorted by relevance
250 StatusOr<py::capsule> BufferToDLPackManagedTensor(py::handle py_buffer, in BufferToDLPackManagedTensor()
1219 Py_buffer py_buffer = {}; in Create() local1240 PythonBuffer(const Py_buffer &py_buffer) : m_buffer(py_buffer) {} in PythonBuffer()
456 auto py_buffer = absl::make_unique<PjRtStreamExecutorBuffer>( in AllocateDestinationBuffer() local