Searched defs:cpu_allocator (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | copy_tensor.cc | 55 void CopyHostToDevice(const Tensor* input, Allocator* cpu_allocator, in CopyHostToDevice()
123 Allocator* cpu_allocator, Allocator* out_allocator, in CopyDeviceToDevice()
225 Allocator* cpu_allocator = src->GetAllocator(host_alloc_attrs); in ViaDMA() local
340 void CopyDeviceToHost(const Tensor* input, Allocator* cpu_allocator, in CopyDeviceToHost()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device_factory.cc | 34 Allocator* gpu_allocator, Allocator* cpu_allocator) in GPUDevice()
68 Allocator* cpu_allocator) override { in CreateGPUDevice()
|
| H A D | gpu_device.cc | 391 Allocator* gpu_allocator, Allocator* cpu_allocator, in BaseGPUDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | allocator.cc | 79 Allocator* cpu_allocator(int numa_node) { in cpu_allocator() function
|
| H A D | cpu_allocator_impl.cc | 193 explicit CPUSubAllocator(CPUAllocator* cpu_allocator) in CPUSubAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | unary_ops_composition_test.cc | 79 Allocator* cpu_allocator = device_->GetAllocator(host_alloc_attrs); in RunComposedOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device.cc | 144 Allocator* cpu_allocator, bool sync_every_op) in PluggableDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | stack.cc | 256 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/ |
| H A D | xla_ops.cc | 538 Allocator* cpu_allocator = ctx->device()->GetAllocator(host_alloc_attrs); in Compute() local
|