| /external/tensorflow/tensorflow/core/common_runtime/ |
| process_function_library_runtime.cc | 75 const string& source_device, const string& target_device,
83 string key = Rendezvous::CreateKey(source_device, src_incarnation,
94 const string& source_device, const string& target_device,
102 string key = Rendezvous::CreateKey(source_device, src_incarnation,
285 string source_device = opts.source_device; local
287 Status s = GetDeviceContext(source_device, &device_context);
293 s = GetDeviceIncarnation(source_device, &src_incarnation);
301 s = SendTensors(source_device, target_device, "arg_", src_incarnation, args,
311 [source_device, target_device, target_incarnation, rendezvous
[all...] |
| process_function_library_runtime_test.cc | 163 opts.source_device = "/job:a/replica:0/task:0/cpu:0";
179 opts.source_device = "/job:a/replica:0/task:0/cpu:0";
196 opts.source_device = "/job:a/replica:0/task:0/cpu:0";
214 opts.source_device = "/job:a/replica:0/task:0/cpu:0";
234 opts.source_device = "/job:a/replica:0/task:0/cpu:0";
|
| process_function_library_runtime.h | 46 // Sends `tensors_to_send` from `source_device` to `target_device` using
52 static Status SendTensors(const string& source_device,
63 // `source_device`) using `rendezvous`. Uses `key_prefix` to construct the
69 const string& source_device, const string& target_device,
|
| function.cc | 674 string source_device = opts.source_device; local
684 s = parent_->GetDeviceIncarnation(source_device, &src_incarnation);
703 // The ProcFLR sends the arguments to the function from the source_device to
706 // to the source_device (caller) so that the ProcFLR can receive them later.
709 source_device, target_device, "arg_", src_incarnation, args.size(),
711 [frame, remote_args, item, source_device, target_device,
726 *exec_args, [item, frame, rets, done, source_device, target_device,
741 target_device, source_device, "ret_", target_incarnation,
[all...] |
| function_test.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| prefetching_kernels.cc | 39 const string& source_device,
46 source_device_(source_device),
176 opts.source_device = source_device_;
180 if (opts.source_device != target_device_) {
268 const string& source_device = ctx->device()->name(); variable
277 [lib, &source_device, &target_device, func_args,
280 lib, func_, buffer_size_, source_device, target_device,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| function_ops.cc | 320 opts.source_device = lib->device()->name();
321 if (opts.source_device != target_device) {
|
| /external/tensorflow/tensorflow/core/framework/ |
| function.h | 487 string source_device = ""; // Fully specified device name. member in struct:tensorflow::FunctionLibraryRuntime::Options
|
