Searched defs:target_device (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 60 def copy_to_device(target_device, source_device="/cpu:0"): argument
89 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 405 FunctionData(const string& target_device, in FunctionData()
412 const string& target_device() { return target_device_; } in target_device() function
|
| D | process_function_library_runtime.cc | 118 const string& source_device, const string& target_device, in SendTensors()
137 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
1034 Device* target_device = nullptr; in GetOutputDevices() local
1339 string target_device; in ReleaseHandle() local
1480 string target_device; in RunInternal() local
|
| D | function.cc | 725 Device* target_device; in IsLocalTarget() local
1001 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device_op_handler.cc | 71 auto target_device = op->DeviceName(); in Execute() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 288 const string& target_device = function_target.first; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.h | 74 string target_device; member
|
| D | generic_layout_optimizer_transposer.cc | 251 absl::string_view target_device, absl::string_view src_format, in AssignDeviceAndDataFormats()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 311 const Device* target_device, EagerContext* ctx, in SerializePackedHandle()
|