Searched defs:target_device (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | prefetching_ops.py | 45 def copy_to_device(target_device, source_device="/cpu:0"): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 59 def copy_to_device(target_device, source_device="/cpu:0"): argument
88 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 255 FunctionData(const string& target_device, in FunctionData()
262 string target_device() { return target_device_; } in target_device() function
|
| D | process_function_library_runtime.cc | 111 const string& source_device, const string& target_device, in SendTensors()
130 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
716 Device* target_device = target_flr->device(); in GetOutputDevices() local
896 string target_device; in ReleaseHandle() local
924 string target_device; in Run() local
|
| D | function.cc | 651 Device* target_device; in IsLocalTarget() local
953 string target_device = parent_->GetDeviceName(handle); in RunRemote() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 311 string target_device; in ComputeAsync() local
|