Searched defs:source_device (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 42 source_device, element_spec): argument
219 source_device="/cpu:0"): argument
411 def __init__(self, devices, source_device, element_spec): argument
471 source_device="/cpu:0", argument
|
| /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/kernels/ |
| D | function_ops.cc | 315 const string& source_device = lib->device()->name(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 122 const string& source_device, const string& target_device, in SendTensors()
141 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
1302 string source_device = opts.source_device; in RunInternal() local
|
| D | function.cc | 1020 string source_device = opts.source_device; in RunRemote() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 700 string source_device = ""; // Fully specified device name. member
|