Home
last modified time | relevance | path

Searched defs:cpu_allocator (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcopy_tensor.cc51 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() local
332 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()
68 Allocator* cpu_allocator) override { in CreateGPUDevice()
Dgpu_device.cc383 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()
/external/tensorflow/tensorflow/core/framework/
Dallocator.cc81 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
Dcpu_allocator_impl.cc156 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc79 Allocator* cpu_allocator = device_->GetAllocator(host_alloc_attrs); in RunComposedOp() local
/external/tensorflow/tensorflow/core/kernels/
Dstack.cc256 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() local
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc424 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local