Searched refs:edge_name (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | copy_tensor.cc | 52 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice() argument
67 edge_name, sync_dst_compute, wrapped_done = std::move(wrapped_done)]( in CopyHostToDevice()
71 CopyHostToDevice(&from, cpu_allocator, out_allocator, edge_name, in CopyHostToDevice()
200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA() argument
208 [&] { return absl::StrCat("#edge_name=", edge_name, "#"); }); in ViaDMA()
209 VLOG(1) << "Copy " << edge_name; in ViaDMA()
257 cpu_tensor, cpu_allocator, out_allocator, edge_name, dst, output, in ViaDMA()
263 CopyHostToDevice(cpu_tensor, cpu_allocator, out_allocator, edge_name, in ViaDMA()
267 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src, in ViaDMA()
276 CopyDeviceToHost(input, cpu_allocator, out_allocator, edge_name, src, in ViaDMA()
[all …]
|
| D | graph_runner.cc | 61 string edge_name(parsed.edge_name); in Send() local
62 if (table_.count(edge_name) > 0) { in Send()
65 table_[edge_name] = val; in Send()
74 string key(parsed.edge_name); in RecvAsync()
|
| D | copy_tensor.h | 43 static void ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context,
73 Allocator* out_allocator, StringPiece edge_name,
|
| D | gpu_device_context.h | 66 void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece edge_name,
|
| D | rendezvous_mgr.cc | 110 parsed.edge_name, send_args.device_context, recv_args.device_context, in SameWorkerRecvDone()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 48 edge_name = StringPiece(buf_.data() + (b.edge_name.data() - b_base), in operator =()
49 b.edge_name.size()); in operator =()
111 out->edge_name = StringPiece(parts[3].data(), parts[3].size()); in ParseKey()
|
| D | rendezvous.h | 65 StringPiece edge_name; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 218 NodeDef* input, const string& edge_name, in MaybeRewriteInput() argument
222 (sa_opti->repeated_outputs().find(edge_name) != in MaybeRewriteInput()
249 VLOG(1) << "Rewrite input " << edge_name << " op " << op->name() in MaybeRewriteInput()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 287 parsed.edge_name, send_args.device_context, recv_args.device_context, in SameWorkerRecvDone()
|