| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 288 const string& target_device = function_target.first; in ComputeAsync() local
298 VLOG(1) << "Instantiating " << func_name << " on " << target_device; in ComputeAsync()
302 " on ", target_device); in ComputeAsync()
312 instantiate_opts.target = target_device; in ComputeAsync()
319 VLOG(1) << "Instantiated " << func_name << " on " << target_device in ComputeAsync()
331 if (opts.source_device != target_device) { in ComputeAsync()
349 VLOG(1) << "Running " << func_name << " on " << target_device in ComputeAsync()
354 ",device=", target_device, "#"); in ComputeAsync()
361 target_device = std::move(function_target.first)](const Status& status) { in ComputeAsync()
365 ",device=", target_device, "#"); in ComputeAsync()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 54 copy_to_device(target_device=device)).prefetch(buffer_size)
60 def copy_to_device(target_device, source_device="/cpu:0"): argument
77 dataset, target_device=target_device,
89 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
98 self._target_device = target_device
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device_op_handler.cc | 71 auto target_device = op->DeviceName(); in Execute() local
72 if (target_device.empty()) { in Execute()
73 target_device = op->GetContext()->HostCPUName(); in Execute()
83 previous, target_device, &new_tesnor)); in Execute()
|
| /external/autotest/client/cros/faft/utils/ |
| D | kernel_handler.py | 66 target_device = self.os_if.get_internal_disk(
69 target_device = self.root_dev
73 'cgpt show %s' % target_device)
81 device = self.os_if.join_part(target_device, line.split()[2])
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 118 const string& source_device, const string& target_device, in SendTensors() argument
127 target_device, name, FrameAndIter(0, 0)); in SendTensors()
137 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
147 target_device, name, FrameAndIter(0, 0)); in ReceiveTensorsAsync()
166 flr = GetFLR(fiter->second->target_device()); in GetRetTypes()
319 if (function_data->target_device() != device_name) { in GetHandleOnDevice()
331 return function_data->target_device(); in GetDeviceName()
1034 Device* target_device = nullptr; in GetOutputDevices() local
1040 target_device = device_set()->FindDeviceByName(target); in GetOutputDevices()
1046 target_device = target_flr->device(); in GetOutputDevices()
[all …]
|
| D | process_function_library_runtime.h | 91 const string& target_device,
105 const string& source_device, const string& target_device,
405 FunctionData(const string& target_device, in FunctionData() argument
408 : target_device_(target_device), in FunctionData()
412 const string& target_device() { return target_device_; } in target_device() function
|
| D | function.cc | 725 Device* target_device; in IsLocalTarget() local
726 if (!device_mgr_->LookupDevice(options.target, &target_device).ok()) { in IsLocalTarget()
731 if (target_device != device_) { in IsLocalTarget()
1001 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
1005 Status s = parent_->GetDeviceContext(target_device, &device_context); in RunRemote()
1012 s.Update(parent_->GetDeviceIncarnation(target_device, &target_incarnation)); in RunRemote()
1051 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
1053 [frame, remote_args, item, source_device, target_device, in RunRemote()
1069 [frame, rets, done, source_device, target_device, in RunRemote()
1083 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 311 const Device* target_device, EagerContext* ctx, in SerializePackedHandle() argument
336 (!ctx->OnSameTask(src_device, target_device)); in SerializePackedHandle()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.h | 59 void AssignDeviceAndDataFormats(absl::string_view target_device,
74 string target_device; member
|
| D | generic_layout_optimizer_transposer.cc | 185 context_->AssignDeviceAndDataFormats(context_->target_device, in ScopedDataFormatUpgrader()
196 context_->AssignDeviceAndDataFormats(context_->target_device, in ~ScopedDataFormatUpgrader()
251 absl::string_view target_device, absl::string_view src_format, in AssignDeviceAndDataFormats() argument
253 this->target_device = string(target_device); in AssignDeviceAndDataFormats()
274 absl::AsciiStrToLower(context.target_device)); in ShouldProcess()
|
| /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 | 133 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/llvm-project/clang/docs/ |
| D | OpenMPSupport.rst | 331 | misc extension | 'target_device' selector in context specifier | :no…
|