Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc336 Device* input_device; in AddInputDevicesToCacheKey() local
850 Device* input_device = tensor_handle->device(); in MaybeUpdateOpDevice() local
/external/tensorflow/tensorflow/python/tpu/
Dtpu_context.py49 input_device=None, argument
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc830 auto& input_device = device_[input_state.device_name]; in MarkCurrNodeExecuted() local
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2379 input_device=None): argument