Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.cc22 bool ExecuteNodeArgs::IsRemote(EagerContext* ctx, Device* input_device, in IsRemote()
37 Device* input_device, in InitPackedHandle()
Dexecute.cc285 Device* input_device; in GetDeviceForInput() local
427 Device* input_device; in GetOrCreateKernelAndDevice() local
856 tensorflow::Device* input_device = input->device(); in EagerRemoteExecute() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc664 string input_device = input->device(); in GroupCrossDeviceControlEdges() local
711 string input_device = input->device(); in GroupCrossDeviceControlEdges() local
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc119 const char* input_device = TFE_TensorHandleDeviceName(input, s); in LoggingDeviceExecute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dinline_function_utils.cc128 const string& input_device = edge->src()->has_assigned_device_name() in InputDevices() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc991 auto& input_device = device_[input_state.device_name]; in MarkNodeExecuted() local
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2389 input_device=None): argument