Searched defs:cpu_allocator (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 33 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()
67 Allocator* cpu_allocator) override { in CreateGPUDevice()
|
| D | gpu_device.cc | 302 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 50 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice()
115 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost()
181 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()
270 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 192 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()
224 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.h | 201 Allocator* cpu_allocator, SYCLDeviceContext* ctx) in SYCLDevice()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 255 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 478 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local
|