| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 75 const string& source_device, const string& target_device, in SendTensors() argument
84 target_device, name, FrameAndIter(0, 0)); in SendTensors()
94 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
103 target_device, name, FrameAndIter(0, 0)); in ReceiveTensorsAsync()
200 if (function_data.target_device != device_name) { in GetHandleOnDevice()
211 return function_data.target_device; in GetDeviceName()
246 string target_device; in ReleaseHandle() local
250 target_device = function_data_[handle].target_device; in ReleaseHandle()
252 flr = GetFLR(target_device); in ReleaseHandle()
271 string target_device; in Run() local
[all …]
|
| D | process_function_library_runtime.h | 53 const string& target_device,
69 const string& source_device, const string& target_device,
149 const string target_device; member
152 FunctionData(const string& target_device, in FunctionData()
154 : target_device(target_device), local_handle(local_handle) {} in FunctionData()
|
| D | function.cc | 465 Device* target_device; in IsLocalTarget() local
466 if (!device_mgr_->LookupDevice(options.target, &target_device).ok()) { in IsLocalTarget()
469 return target_device == device_; in IsLocalTarget()
673 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
677 Status s = parent_->GetDeviceContext(target_device, &device_context); in RunRemote()
685 s.Update(parent_->GetDeviceIncarnation(target_device, &target_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()
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | prefetching_ops.py | 27 target_device, argument
36 target_device=target_device,
|
| /external/autotest/client/cros/faft/utils/ |
| D | kernel_handler.py | 63 target_device = self.os_if.get_internal_disk(
66 target_device = self.root_dev
70 'cgpt show %s' % target_device)
78 device = self.os_if.join_part(target_device, line.split()[2])
|
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| D | prefetching_kernels.cc | 40 const string& target_device, in FunctionBufferingResource() argument
47 target_device_(target_device), in FunctionBufferingResource()
261 const string& target_device = 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 | 298 const string& target_device = in ComputeAsync() local
307 instantiate_opts.target = target_device; in ComputeAsync()
321 if (opts.source_device != target_device) { in ComputeAsync()
|
| /external/tensorflow/tensorflow/contrib/data/python/kernel_tests/ |
| D | prefetching_ops_test.py | 66 target_device=target,
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | datasets.py | 118 target_device=target,
|