Home
last modified time | relevance | path

Searched refs:input_device_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc337 DeviceNameUtils::ParsedName input_device_name; in InputTFE_OutputTensorHandles() local
338 if (!DeviceNameUtils::ParseFullOrLocalName(device_name, &input_device_name)) { in InputTFE_OutputTensorHandles()
346 return DeviceNameUtils::AreCompatibleDevNames(input_device_name, in InputTFE_OutputTensorHandles()
533 tensorflow::DeviceNameUtils::ParsedName input_device_name; in PYBIND11_MODULE() local
535 device_name, &input_device_name)) { in PYBIND11_MODULE()
549 input_device_name, device->parsed_name())) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc858 const string* input_device_name = &input_device_or_cpu->name(); in EagerRemoteExecute() local
884 input_device_name = &remote_cpu_device->name(); in EagerRemoteExecute()
904 *input_device_name, serialize_resource_dtype_and_shape)); in EagerRemoteExecute()