Home
last modified time | relevance | path

Searched refs:has_replica (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc119 p->has_replica = !absl::ConsumePrefix(&fullname, "*"); in ParseFullName()
120 if (p->has_replica && !ConsumeNumber(&fullname, &p->replica)) { in ParseFullName()
190 if (!parsed_name->has_replica) { in CompleteName()
192 parsed_name->has_replica = true; in CompleteName()
220 if (!(parsed_basename.has_job && parsed_basename.has_replica && in CanonicalizeDeviceName()
247 if (pn.has_replica) strings::StrAppend(&buf, "/replica:", pn.replica); in ParsedNameToString()
267 if (less_specific.has_replica && in IsSpecification()
268 (!more_specific.has_replica || in IsSpecification()
293 if (a.has_replica && b.has_replica && (a.replica != b.replica)) { in AreCompatibleDevNames()
314 if (less_specific.has_replica) { in EnsureSpecification()
[all …]
Ddevice_name_utils.h55 has_replica = false; in Clear()
68 (has_replica ? (other.has_replica && replica == other.replica)
69 : !other.has_replica) &&
83 bool has_replica = false; member
119 return name.has_job || name.has_replica || name.has_task || name.has_type || in HasSomeDetails()
Ddevice_name_utils_test.cc93 EXPECT_TRUE(p.has_replica); in TEST()
111 EXPECT_TRUE(p.has_replica); in TEST()
127 EXPECT_TRUE(p.has_replica); in TEST()
141 EXPECT_TRUE(p.has_replica); in TEST()
153 EXPECT_TRUE(p.has_replica); in TEST()
165 EXPECT_TRUE(p.has_replica); in TEST()
177 EXPECT_TRUE(p.has_replica); in TEST()
189 EXPECT_FALSE(p.has_replica); in TEST()
199 EXPECT_TRUE(p.has_replica); in TEST()
211 EXPECT_TRUE(p.has_replica); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device.cc105 !worker_name_parsed.has_job || !worker_name_parsed.has_replica || in NewRemoteDevices()
Dmaster.cc336 (!x.has_replica || !y.has_replica || x.replica == y.replica) && in Intersects()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc28 !parsed.has_job || !parsed.has_replica || !parsed.has_task || in RegisterCustomDevice()
Dcontext.cc1383 pw.has_replica = true; in SetRemoteDeviceFilters()
1439 (!pn.has_replica || !pf.has_replica || pn.replica == pf.replica) && in FilterDevicesForRemoteWorkers()
Deager_operation.cc468 if (!device_parsed_name_.has_job && !device_parsed_name_.has_replica && in IsLocal()
/external/tensorflow/tensorflow/core/tfrt/runtime/
Druntime.cc61 !device_parsed_name.has_replica || in InitializeOpHandlers()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc180 } else if (!name.has_job || !name.has_replica || !name.has_task || in AddDevice()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.cc131 TF_RET_CHECK(parsed_name.has_replica); in GetHostSystemDevices()
Ddistributed_tpu_rewrite_pass.cc2782 if (var_device.has_job && var_device.has_replica && var_device.has_task && in FindVariableInputs()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_channel.cc300 if (!parsed.has_replica || parsed.replica != 0) { in TranslateTask()