| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 75 const string& source_device, const string& target_device, in SendTensors() argument
83 string key = Rendezvous::CreateKey(source_device, src_incarnation, in SendTensors()
94 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
102 string key = Rendezvous::CreateKey(source_device, src_incarnation, in ReceiveTensorsAsync()
285 string source_device = opts.source_device; in Run() local
287 Status s = GetDeviceContext(source_device, &device_context); in Run()
293 s = GetDeviceIncarnation(source_device, &src_incarnation); in Run()
301 s = SendTensors(source_device, target_device, "arg_", src_incarnation, args, in Run()
311 [source_device, target_device, target_incarnation, rendezvous, in Run()
322 ReceiveTensorsAsync(target_device, source_device, "ret_", in Run()
|
| D | process_function_library_runtime_test.cc | 163 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
179 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
196 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
214 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
234 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
|
| D | process_function_library_runtime.h | 52 static Status SendTensors(const string& source_device,
69 const string& source_device, const string& target_device,
|
| D | function.cc | 674 string source_device = opts.source_device; in RunRemote() local
684 s = parent_->GetDeviceIncarnation(source_device, &src_incarnation); in RunRemote()
709 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
711 [frame, remote_args, item, source_device, target_device, in RunRemote()
726 *exec_args, [item, frame, rets, done, source_device, target_device, in RunRemote()
741 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| D | function_test.cc | 1248 opts.source_device = "/device:CPU:1"; in TEST_F()
1256 opts.source_device = "/job:localhost/replica:0/task:0/cpu:2"; in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| D | prefetching_kernels.cc | 39 const string& source_device, in FunctionBufferingResource() argument
46 source_device_(source_device), in FunctionBufferingResource()
176 opts.source_device = source_device_; in FillBuffer()
180 if (opts.source_device != target_device_) { in FillBuffer()
268 const string& source_device = ctx->device()->name(); in Compute() local
277 [lib, &source_device, &target_device, func_args, in Compute()
280 lib, func_, buffer_size_, source_device, target_device, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 320 opts.source_device = lib->device()->name(); in ComputeAsync()
321 if (opts.source_device != target_device) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 487 string source_device = ""; // Fully specified device name. member
|