Searched refs:allocated_buffer (Results 1 – 8 of 8) sorted by relevance
99 PyObject *allocated_buffer; member109 PyObject *allocated_buffer; member659 self->allocated_buffer = NULL; in _overlapped_Overlapped_impl()676 Py_CLEAR(self->allocated_buffer); in Overlapped_clear()687 if(self->read_from.allocated_buffer) { in Overlapped_clear()688 Py_CLEAR(self->read_from.allocated_buffer); in Overlapped_clear()917 self->read_from.allocated_buffer != NULL)) in _overlapped_Overlapped_getresult_impl()933 assert(PyBytes_CheckExact(self->allocated_buffer)); in _overlapped_Overlapped_getresult_impl()934 if (transferred != PyBytes_GET_SIZE(self->allocated_buffer) && in _overlapped_Overlapped_getresult_impl()935 _PyBytes_Resize(&self->allocated_buffer, transferred)) in _overlapped_Overlapped_getresult_impl()[all …]
75 return storage_->allocated_buffer; in GetStorage()89 if (!is_inline) allocated_buffer = AlignedAlloc(alignment, size); in Storage()93 if (!is_inline) AlignedFree(allocated_buffer); in ~Storage()104 void* allocated_buffer; member
78 OwningDeviceMemory allocated_buffer, in AllocateBytes()85 DeviceMemory<uint8> allocated_buffer_memory(*allocated_buffer); in AllocateBytes()115 allocated_buffers_.emplace_back(std::move(allocated_buffer), byte_size); in AllocateBytes()
163 auto allocated_buffer, in AllocateOutputMemoryWithInputReuse()169 result_buffer = allocated_buffer.Release(); in AllocateOutputMemoryWithInputReuse()
217 auto allocated_buffer, in AllocateOutputMemoryWithInputReuse()221 result_buffer = allocated_buffer.Release(); in AllocateOutputMemoryWithInputReuse()
289 se::OwningDeviceMemory allocated_buffer, in CreateResultShapedBuffer()292 result_buffer = allocated_buffer.Release(); in CreateResultShapedBuffer()
797 StatusOr<se::OwningDeviceMemory> allocated_buffer = in ExecuteAsyncOnStreamImpl() local799 if (!allocated_buffer.ok()) { in ExecuteAsyncOnStreamImpl()801 allocated_buffer.status().error_message(), in ExecuteAsyncOnStreamImpl()804 result_buffer = allocated_buffer->Release(); in ExecuteAsyncOnStreamImpl()
99 TF_ASSIGN_OR_RETURN(se::OwningDeviceMemory allocated_buffer, in AllocateBytes()104 se::DeviceMemoryBase buffer_addr = *allocated_buffer; in AllocateBytes()105 allocated_buffers_.push_back(std::move(allocated_buffer)); in AllocateBytes()