Lines Matching defs:cpu_allocator
50 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator,
55 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
108 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator,
113 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
167 Allocator* cpu_allocator, Allocator* out_allocator,
175 Tensor copy(cpu_allocator, DT_VARIANT, input->shape());
255 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs);
266 CopyDeviceToDevice(ri.copy_function, cpu_allocator, out_allocator,
281 new Tensor(cpu_allocator, input->dtype(), input->shape());
291 [delete_and_done, recv_dev_context, cpu_tensor, cpu_allocator,
299 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name,
304 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src,
313 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src,
321 CopyHostToDevice(input, cpu_allocator, out_allocator, edge_name, dst,
