| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 59 def copy_to_device(target_device, source_device="/cpu:0"): argument
77 source_device=source_device).with_options(options)
88 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
100 self._source_device_string = source_device
101 self._source_device = ops.convert_to_tensor(source_device)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 38 source_device, element_structure): argument
56 target=source_device,
91 target=source_device,
119 target=source_device,
198 source_device="/cpu:0"): argument
218 self._source_device = source_device
219 self._source_device_tensor = ops.convert_to_tensor(source_device)
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | prefetching_ops.py | 45 def copy_to_device(target_device, source_device="/cpu:0"): argument
56 return prefetching_ops.copy_to_device(target_device, source_device)
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 111 const string& source_device, const string& target_device, in SendTensors() argument
119 string key = Rendezvous::CreateKey(source_device, src_incarnation, in SendTensors()
130 const string& source_device, const string& target_device, in ReceiveTensorsAsync() argument
138 string key = Rendezvous::CreateKey(source_device, src_incarnation, in ReceiveTensorsAsync()
949 string source_device = opts.source_device; in Run() local
951 Status s = GetDeviceContext(source_device, &device_context); in Run()
957 s = GetDeviceIncarnation(source_device, &src_incarnation); in Run()
965 s = SendTensors(source_device, target_device, "arg_", src_incarnation, args, in Run()
976 [source_device, target_device, target_incarnation, rendezvous, in Run()
988 ReceiveTensorsAsync(target_device, source_device, "ret_", in Run()
|
| D | process_function_library_runtime_test.cc | 270 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
286 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
303 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
321 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
341 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
363 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
391 opts.source_device = "/job:a/replica:0/task:0/cpu:0"; in TEST_F()
|
| D | process_function_library_runtime.h | 56 static Status SendTensors(const string& source_device,
71 const string& source_device, const string& target_device,
|
| D | function.cc | 954 string source_device = opts.source_device; in RunRemote() local
963 s = parent_->GetDeviceIncarnation(source_device, &src_incarnation); in RunRemote()
1003 source_device, target_device, "arg_", src_incarnation, args.size(), in RunRemote()
1005 [frame, remote_args, item, source_device, target_device, in RunRemote()
1021 [frame, rets, done, source_device, target_device, in RunRemote()
1035 target_device, source_device, "ret_", target_incarnation, in RunRemote()
|
| D | function_test.cc | 1592 opts.source_device = "/device:CPU:1"; in TEST_F()
1600 opts.source_device = "/job:localhost/replica:0/task:0/cpu:2"; in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 308 const string& source_device = lib->device()->name(); in ComputeAsync() local
315 source_device, &target_device), in ComputeAsync()
352 opts.source_device = source_device; in ComputeAsync()
353 if (opts.source_device != target_device) { in ComputeAsync()
|
| D | partitioned_function_ops.cc | 223 run_opts.source_device = in RunFunction()
|
| /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 | 105 …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 | 422 prefetching_ops.copy_to_device("/gpu:0", source_device="/cpu:0"))
424 prefetching_ops.copy_to_device("/cpu:0", source_device="/gpu:0"))
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 627 string source_device = ""; // Fully specified device name. member
|