Searched defs:host_memory_allocator (Results 1 – 7 of 7) sorted by relevance
39 Allocator* host_memory_allocator) in GPUDeviceContext()63 Allocator* host_memory_allocator() const override { in host_memory_allocator() function
123 virtual Allocator* host_memory_allocator() const { return nullptr; } in host_memory_allocator() function
437 auto host_memory_allocator = in GetGpuClient() local
279 tensorflow::Allocator* host_memory_allocator() const { in host_memory_allocator() function
209 std::unique_ptr<tensorflow::Allocator> host_memory_allocator, in PjRtStreamExecutorClient()
341 Allocator* host_memory_allocator = device_context->host_memory_allocator(); in CopyCPUTensorToGPU() local
453 Allocator* host_memory_allocator = GetAllocator(attr); in Init() local