Home
last modified time | relevance | path

Searched refs:GetSystemDevice (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.h39 static Status GetSystemDevice(const string& system_spec_string,
Ddistributed_tpu_rewrite_helpers.cc29 Status DistributedTPURewriteHelpers::GetSystemDevice( in GetSystemDevice() function in tensorflow::DistributedTPURewriteHelpers
222 GetSystemDevice(spec_string, device_set, &spec, &device)); in ForConfigurationNodeMatchingType()
Ddistributed_tpu_rewrite_pass.cc1368 TF_RETURN_IF_ERROR(DistributedTPURewriteHelpers::GetSystemDevice( in GetTPUDeviceNames()