Searched defs:cpu_allocator (Results 1 – 8 of 8) sorted by relevance
51 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice()121 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()223 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA() local332 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost()
34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()68 Allocator* cpu_allocator) override { in CreateGPUDevice()
383 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()
81 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
156 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()
79 Allocator* cpu_allocator = device_->GetAllocator(host_alloc_attrs); in RunComposedOp() local
256 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() local
424 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local