Searched refs:FindMatchingDevices (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
D | distributed_tpu_rewrite_helpers.cc | 47 device_set.FindMatchingDevices(*system_spec, &system_devices); in GetSystemDevice() 114 device_set.FindMatchingDevices(host_spec, host_system_devices); in GetHostSystemDevices() 177 device_set.FindMatchingDevices(device_spec, &host_tpu_devices); in GetTPUDevices()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | device_set.h | 58 void FindMatchingDevices(const DeviceNameUtils::ParsedName& spec,
|
D | device_set.cc | 44 void DeviceSet::FindMatchingDevices(const DeviceNameUtils::ParsedName& spec, in FindMatchingDevices() function in tensorflow::DeviceSet
|
D | colocation_graph.cc | 1115 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates() 1138 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates() 1189 device_set_.FindMatchingDevices(root_member.requested_device_name(), in GetDevicesForNode() 1218 device_set_.FindMatchingDevices(specified_device_name, in GetDevicesForNode()
|
D | process_function_library_runtime.cc | 496 device_set.FindMatchingDevices(parsed, &matching_devices); in PinArgsAndRets()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | tpu_rewrite_device_util.cc | 64 void FindMatchingDevices(Devices devices, const Device& spec, in FindMatchingDevices() function 92 FindMatchingDevices(devices, spec, &system_devices); in GetTPUSystemDevices() 133 FindMatchingDevices(devices, device_spec, &host_tpu_devices); in GetTPUDevices()
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_graph.cc | 659 device_set->FindMatchingDevices(parsed_name, &devices); in GetDeviceAndAllocator()
|