Searched defs:cpu_allocator (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 51 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice()
121 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()
222 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA() local
331 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()
68 Allocator* cpu_allocator) override { in CreateGPUDevice()
|
| D | gpu_device.cc | 318 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 80 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
|
| D | cpu_allocator_impl.cc | 157 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.h | 201 Allocator* cpu_allocator, SYCLDeviceContext* ctx) in SYCLDevice()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 73 Allocator* cpu_allocator = device_->GetAllocator(host_alloc_attrs); in RunComposedOp() local
|
| /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 | 521 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local
|