| /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)
411 def __init__(self, devices, source_device, element_spec): argument
413 self._source_device = source_device
443 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 | 315 const string& source_device = lib->device()->name(); in ComputeAsync() local
322 source_device, &target_device), in ComputeAsync()
371 opts.source_device = source_device; in ComputeAsync()
372 if (opts.source_device != target_device) { in ComputeAsync()
|
| D | partitioned_function_ops.cc | 244 run_opts.source_device = in RunFunction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 341 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
356 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
372 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
389 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
408 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
429 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
459 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
982 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
998 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
1024 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
|
| D | process_function_library_runtime.cc | 122 const string& source_device, const string& target_device, in SendTensors() argument
130 string key = Rendezvous::CreateKey(source_device, src_incarnation, in SendTensors()
141 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
150 string key = Rendezvous::CreateKey(source_device, src_incarnation, in ReceiveTensorsAsync()
1302 string source_device = opts.source_device; in RunInternal() local
1304 Status s = GetDeviceContext(source_device, &device_context); in RunInternal()
1310 s = GetDeviceIncarnation(source_device, &src_incarnation); in RunInternal()
1318 s = SendTensors(source_device, target_device, "arg_", src_incarnation, in RunInternal()
1329 [source_device, target_device, target_incarnation, rendezvous, in RunInternal()
1340 ReceiveTensorsAsync(target_device, source_device, "ret_", in RunInternal()
|
| D | process_function_library_runtime.h | 89 static Status SendTensors(const string& source_device,
104 const string& source_device, const string& target_device,
|
| D | function.cc | 1020 string source_device = opts.source_device; in RunRemote() local
1029 s = parent_->GetDeviceIncarnation(source_device, &src_incarnation); in RunRemote()
1069 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
1071 [frame, remote_args, item, source_device, target_device, in RunRemote()
1087 [frame, rets, done, source_device, target_device, in RunRemote()
1101 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| D | function_test.cc | 1858 opts.source_device = "/device:CPU:1"; in TEST_F()
1866 opts.source_device = "/job:localhost/replica:0/task:0/cpu:2"; in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 93 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | copy_to_device_test.py | 424 prefetching_ops.copy_to_device("/gpu:0", source_device="/cpu:0"))
426 prefetching_ops.copy_to_device("/cpu:0", source_device="/gpu:0"))
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 121 …argspec: "args=[\'target_device\', \'source_device\'], varargs=None, keywords=None, defaults=[\'/c…
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 700 string source_device = ""; // Fully specified device name. member
|
| D | function.cc | 1723 " remote_execution=", remote_execution, " source_device=", source_device, in DebugString()
|