| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 42 source_device, element_spec): argument
60 target=source_device,
96 target=source_device,
124 target=source_device,
219 source_device="/cpu:0"): argument
238 self._source_device = source_device
239 self._source_device_tensor = ops.convert_to_tensor(source_device)
415 def __init__(self, devices, source_device, element_spec): argument
417 self._source_device = source_device
447 source_device=self._source_device,
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 60 def copy_to_device(target_device, source_device="/cpu:0"): argument
78 source_device=source_device).with_options(options)
89 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
101 self._source_device_string = source_device
102 self._source_device = ops.convert_to_tensor(source_device)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 276 const string& source_device = lib->device()->name(); in ComputeAsync() local
284 target->scalar<tstring>()(), source_device, &function_target.first), in ComputeAsync()
330 opts.source_device = source_device; in ComputeAsync()
331 if (opts.source_device != target_device) { in ComputeAsync()
|
| D | partitioned_function_ops.cc | 251 run_opts.source_device = in RunFunction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 382 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
396 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
414 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
430 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
448 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
490 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
519 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
1162 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
1177 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
1224 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
|
| D | process_function_library_runtime.cc | 118 const string& source_device, const string& target_device, in SendTensors() argument
126 string key = Rendezvous::CreateKey(source_device, src_incarnation, in SendTensors()
137 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
146 string key = Rendezvous::CreateKey(source_device, src_incarnation, in ReceiveTensorsAsync()
1505 string source_device = opts.source_device; in RunInternal() local
1507 Status s = GetDeviceContext(source_device, &device_context); in RunInternal()
1513 s = GetDeviceIncarnation(source_device, &src_incarnation); in RunInternal()
1523 s = SendTensors(source_device, target_device, "arg_", src_incarnation, in RunInternal()
1534 [source_device, target_device, target_incarnation, rendezvous, in RunInternal()
1546 ReceiveTensorsAsync(target_device, source_device, "ret_", in RunInternal()
|
| D | function.cc | 1002 string source_device = opts.source_device; in RunRemote() local
1011 s = parent_->GetDeviceIncarnation(source_device, &src_incarnation); in RunRemote()
1051 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
1053 [frame, remote_args, item, source_device, target_device, in RunRemote()
1069 [frame, rets, done, source_device, target_device, in RunRemote()
1083 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| D | process_function_library_runtime.h | 90 static Status SendTensors(const string& source_device,
105 const string& source_device, const string& target_device,
|
| D | function_test.cc | 1985 opts.source_device = "/device:CPU:1"; in TEST_F()
1993 opts.source_device = "/job:localhost/replica:0/task:0/cpu:2"; in TEST_F()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | copy_to_device_test.py | 406 prefetching_ops.copy_to_device("/gpu:0", source_device="/cpu:0"))
408 prefetching_ops.copy_to_device("/cpu:0", source_device="/gpu:0"))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 105 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 133 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 799 std::string source_device = ""; // Fully specified device name. member
|
| D | function.cc | 1765 " remote_execution=", remote_execution, " source_device=", source_device, in DebugString()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 1767 self._dataset, self._devices, source_device=host_device)
|