| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 111 const string& source_device, const string& target_device, in SendTensors() argument
120 target_device, name, FrameAndIter(0, 0)); in SendTensors()
130 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
139 target_device, name, FrameAndIter(0, 0)); in ReceiveTensorsAsync()
253 if (function_data->target_device() != device_name) { in GetHandleOnDevice()
265 return function_data->target_device(); in GetDeviceName()
716 Device* target_device = target_flr->device(); in GetOutputDevices() local
724 (*output_devices)[ret_index] = target_device; in GetOutputDevices()
727 comp_data.ret_alloc_attrs_[j].on_host() ? nullptr : target_device; in GetOutputDevices()
896 string target_device; in ReleaseHandle() local
[all …]
|
| D | process_function_library_runtime.h | 57 const string& target_device,
71 const string& source_device, const string& target_device,
255 FunctionData(const string& target_device, in FunctionData() argument
258 : target_device_(target_device), in FunctionData()
262 string target_device() { return target_device_; } in target_device() function
|
| D | function.cc | 651 Device* target_device; in IsLocalTarget() local
652 if (!device_mgr_->LookupDevice(options.target, &target_device).ok()) { in IsLocalTarget()
657 if (target_device != device_) { in IsLocalTarget()
953 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
957 Status s = parent_->GetDeviceContext(target_device, &device_context); in RunRemote()
964 s.Update(parent_->GetDeviceIncarnation(target_device, &target_incarnation)); in RunRemote()
1003 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
1005 [frame, remote_args, item, source_device, target_device, in RunRemote()
1021 [frame, rets, done, source_device, target_device, in RunRemote()
1035 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 311 string target_device; in ComputeAsync() local
315 source_device, &target_device), in ComputeAsync()
320 instantiate_opts.target = target_device; in ComputeAsync()
322 FunctionTarget function_target = {target_device, lib}; in ComputeAsync()
331 VLOG(1) << "Instantiating " << func_.name() << " on " << target_device; in ComputeAsync()
333 "RemoteCall: Instantiate: ", func_.name(), " on ", target_device)); in ComputeAsync()
341 VLOG(1) << "Instantiated " << func_.name() << " on " << target_device in ComputeAsync()
353 if (opts.source_device != target_device) { in ComputeAsync()
378 "RemoteCall: Run: ", func_.name(), " on ", target_device)); in ComputeAsync()
379 VLOG(1) << "Running " << func_.name() << " on " << target_device in ComputeAsync()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 53 copy_to_device(target_device=device)).prefetch(buffer_size)
59 def copy_to_device(target_device, source_device="/cpu:0"): argument
76 dataset, target_device=target_device,
88 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
97 self._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/python/ops/ |
| D | prefetching_ops.py | 45 def copy_to_device(target_device, source_device="/cpu:0"): argument
56 return prefetching_ops.copy_to_device(target_device, source_device)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 105 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 105 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|