Searched defs:target_device (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 66 def copy_to_device(target_device, source_device="/cpu:0"): argument
91 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 461 FunctionData(const string& target_device, in FunctionData()
468 const string& target_device() { return target_device_; } in target_device() function
|
| D | process_function_library_runtime.cc | 121 const string& source_device, const string& target_device, in SendTensors()
140 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
1177 Device* target_device = nullptr; in GetOutputDevices() local
1603 string target_device; in ReleaseHandle() local
1770 string target_device; in RunInternal() local
|
| D | function.cc | 730 Device* target_device; in IsLocalTarget() local
1036 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/tpu/kernels/ |
| D | tpu_functional_ops.h | 280 Status ShouldUseRemoteExecutionForFn(const std::string& target_device, in ShouldUseRemoteExecutionForFn()
|
| D | tpu_functional_ops.cc | 1597 const string& target_device = entry.device; in InitializeShardedVarOnTPU() local
2463 const string& target_device, FHandle* handle, in InstantiatePartition()
2719 const string& target_device = entry.device; in ExecuteFunctions() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.h | 79 string target_device; member
|
| D | generic_layout_optimizer_transposer.cc | 249 absl::string_view target_device, absl::string_view src_format, in AssignDeviceAndDataFormats()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 303 const string& target_device = function_target.first; in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 320 const Device* target_device, EagerContext* ctx, in SerializePackedHandle()
|