Home
last modified time | relevance | path

Searched defs:source_device (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/data/python/ops/
Dprefetching_ops.py45 def copy_to_device(target_device, source_device="/cpu:0"): argument
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py59 def copy_to_device(target_device, source_device="/cpu:0"): argument
88 def __init__(self, input_dataset, target_device, source_device="/cpu:0"): argument
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py38 source_device, element_structure): argument
198 source_device="/cpu:0"): argument
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc308 const string& source_device = lib->device()->name(); in ComputeAsync() local
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc111 const string& source_device, const string& target_device, in SendTensors()
130 const string& source_device, const string& target_device, in ReceiveTensorsAsync()
949 string source_device = opts.source_device; in Run() local
Dfunction.cc954 string source_device = opts.source_device; in RunRemote() local
/external/tensorflow/tensorflow/core/framework/
Dfunction.h627 string source_device = ""; // Fully specified device name. member