Searched refs:FindMatchingDevices (Results 1 – 9 of 9) 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()
|
| D | update_tpu_embedding_ops_passes.cc | 59 options.device_set->FindMatchingDevices(tpu_device_spec, &tpu_devices); in Run()
96 options.device_set->FindMatchingDevices(single_tpu_device_spec, in Run()
|
| /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 | 1183 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1206 device_set_.FindMatchingDevices(soft_device_name, possible_devices); in GetSoftDeviceCandidates()
1257 device_set_.FindMatchingDevices(root_member.requested_device_name(), in GetDevicesForNode()
1286 device_set_.FindMatchingDevices(specified_device_name, in GetDevicesForNode()
|
| D | process_function_library_runtime.cc | 539 device_set.FindMatchingDevices(parsed, &matching_devices); in PinArgsAndRets()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | tpu_rewrite_device_util.cc | 60 void FindMatchingDevices(Devices devices, const Device& spec, in FindMatchingDevices() function
88 FindMatchingDevices(devices, spec, &system_devices); in GetTPUSystemDevices()
129 FindMatchingDevices(devices, device_spec, &host_tpu_devices); in GetTPUDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 670 device_set->FindMatchingDevices(parsed_name, &devices); in GetDeviceAndAllocator()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 1220 device_set_.FindMatchingDevices(tpu_device_name, &tpu_devices_); in ComputeAsync()
|