| /external/tensorflow/tensorflow/core/common_runtime/ |
| copy_tensor.cc | 280 Tensor* cpu_tensor = local
283 [cpu_tensor](StatusCallback done_,
286 delete cpu_tensor;
291 [delete_and_done, recv_dev_context, cpu_tensor, cpu_allocator,
299 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name,
305 cpu_tensor, send_dev_context,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| stack_ops.cc | 295 Tensor* cpu_tensor = variable
298 &tensor, "StackPush", device, cpu_tensor,
299 [cpu_tensor, stack, ctx, done](const Status& s) {
303 ctx->SetStatus(stack->Push({*cpu_tensor, alloc_attrs, true}));
306 ctx->set_output(0, *cpu_tensor);
309 delete cpu_tensor;
414 Tensor* cpu_tensor = &value.tensor; variable
417 new Tensor(gpu_allocator, cpu_tensor->dtype(), cpu_tensor->shape());
419 cpu_tensor, device, device_tensor
[all...] |
| dynamic_partition_op_gpu.cu.cc | 279 Tensor cpu_tensor; local
286 &cpu_tensor, alloc_attr),
292 ->ThenMemcpy(cpu_tensor.flat<int32>().data(), wrapped,
304 partition_ref, cpu_tensor, done]() {
306 this->AllocateOutputs(c, &data, &partitions, &cpu_tensor, &outputs, done);
|
