Home
last modified time | relevance | path

Searched refs:GetTPUDevices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.h61 static Status GetTPUDevices(const DeviceNameUtils::ParsedName& system_spec,
Ddistributed_tpu_rewrite_helpers.cc150 Status DistributedTPURewriteHelpers::GetTPUDevices( in GetTPUDevices() function in tensorflow::DistributedTPURewriteHelpers
Ddistributed_tpu_rewrite_pass.cc1376 TF_RETURN_IF_ERROR(DistributedTPURewriteHelpers::GetTPUDevices( in GetTPUDeviceNames()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc121 Status GetTPUDevices( in GetTPUDevices() function
459 TF_RETURN_IF_ERROR(GetTPUDevices(devices, system_devices, &tpu_devices)); in GetTPUCompilationAndExecutionDevices()