Home
last modified time | relevance | path

Searched refs:host_memory_allocator (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_util.cc341 Allocator* host_memory_allocator = device_context->host_memory_allocator(); in CopyCPUTensorToGPU() local
353 if (host_memory_allocator == nullptr) { in CopyCPUTensorToGPU()
363 staging_buffer = host_memory_allocator->AllocateRaw( in CopyCPUTensorToGPU()
379 host_memory_allocator]() { in CopyCPUTensorToGPU()
381 host_memory_allocator->DeallocateRaw(staging_buffer); in CopyCPUTensorToGPU()
H A Dgpu_device.cc453 Allocator* host_memory_allocator = GetAllocator(attr); in Init() local
461 stream_->device_to_device, host_memory_allocator); in Init()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgpu_device_context.h39 Allocator* host_memory_allocator) in GPUDeviceContext() argument
48 host_memory_allocator_(host_memory_allocator) { in GPUDeviceContext()
63 Allocator* host_memory_allocator() const override { in host_memory_allocator() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc209 std::unique_ptr<tensorflow::Allocator> host_memory_allocator, in PjRtStreamExecutorClient() argument
215 host_memory_allocator_(std::move(host_memory_allocator)), in PjRtStreamExecutorClient()
752 void* staging_buffer = host_memory_allocator()->AllocateRaw( in BufferFromHostBuffer()
759 on_delete_callback = [staging_buffer, host_memory_allocator = in BufferFromHostBuffer()
760 host_memory_allocator()]() { in BufferFromHostBuffer()
761 host_memory_allocator->DeallocateRaw(staging_buffer); in BufferFromHostBuffer()
792 void* ptr = host_memory_allocator()->AllocateRaw( in BufferFromHostBuffer()
795 ptr, [host_memory_allocator = host_memory_allocator()](void* ptr) { in BufferFromHostBuffer()
796 host_memory_allocator->DeallocateRaw(ptr); in BufferFromHostBuffer()
H A Dgpu_device.cc437 auto host_memory_allocator = in GetGpuClient() local
453 std::move(host_memory_allocator), in GetGpuClient()
H A Dpjrt_stream_executor_client.h152 std::unique_ptr<tensorflow::Allocator> host_memory_allocator,
279 tensorflow::Allocator* host_memory_allocator() const { in host_memory_allocator() function
/aosp_15_r20/device/generic/goldfish-opengl/system/gralloc/
Dgralloc_old.cpp226 : host_memory_allocator( in gralloc_dmaregion_t()
234 host_memory_allocator.hostMalloc(&address_space_block, sz); in gralloc_dmaregion_t()
238 GoldfishAddressSpaceHostMemoryAllocator host_memory_allocator; member
288 grdma->host_memory_allocator.hostFree(&grdma->address_space_block); in put_gralloc_region_direct_mem_locked()
330 GoldfishAddressSpaceHostMemoryAllocator* allocator = &grdma->host_memory_allocator; in gralloc_dmaregion_register_ashmem_direct_mem_locked()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice_base.h123 virtual Allocator* host_memory_allocator() const { return nullptr; } in host_memory_allocator() function