Searched defs:source_device (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 40 source_device, element_spec, iterator_is_anonymous): argument
227 source_device="/cpu:0"): argument
372 def __init__(self, devices, source_device, element_spec): argument
431 source_device="/cpu:0", argument
|
| /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/kernels/ |
| D | function_ops.cc | 291 const string& source_device = lib->device()->name(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| 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()
1795 string source_device = opts.source_device; in RunInternal() local
|
| D | function.cc | 1037 string source_device = opts.source_device; in RunRemote() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 858 std::string source_device = ""; // Fully specified device name. member
|