| /external/tensorflow/tensorflow/core/debug/ |
| debug_gateway.cc | 83 Allocator* cpu_allocator = tensorflow::cpu_allocator(); local
84 Tensor cpu_tensor(cpu_allocator, src_tensor->dtype(), src_tensor->shape());
|
| /external/tensorflow/tensorflow/core/framework/ |
| allocator.cc | 127 Allocator* cpu_allocator() { function in namespace:tensorflow
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| copy_tensor.cc | 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); local
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,
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| stack_ops.cc | 294 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); variable
296 new Tensor(cpu_allocator, tensor.dtype(), tensor.shape());
|
