Searched refs:Replicated (Results 1 – 3 of 3) sorted by relevance
102 struct Replicated { struct104 bool operator==(const Replicated& other) const { argument107 bool operator!=(const Replicated& other) const {112 using MeshDimAssignment = absl::variant<ShardedAxis, Replicated>;
390 py::class_<Replicated> replicated(pmap_lib, "Replicated"); in BuildPmapSubmodule()392 .def_readonly("replicas", &Replicated::replicas) in BuildPmapSubmodule()394 [](const Replicated& x) { in BuildPmapSubmodule()397 .def("__eq__", [](const Replicated& self, const Replicated& other) { in BuildPmapSubmodule()
275 "### Replicated model execution\n",