Searched defs:replicated (Results 1 – 12 of 12) sorted by relevance
357 explicit HloSharding(bool manual, bool replicated, in HloSharding()
105 auto* replicated = proto.add_tuple_shardings(); in TEST_F() local
1211 const auto& replicated = in ParseInstructionRhs() local3045 bool replicated = false; in ParseSingleSharding() local
548 const bool replicated = tpu_devices.size() != 1; in AddToParallelExecuteOp() local590 const bool replicated = tpu_devices.size() != 1; in AssignDevicesToReplicatedExecute() local
677 def replicated(mesh: Mesh, rank: int) -> 'Layout': member in Layout
146 replicated = 0 variable in Uop
120 nir_def *replicated[4]; in pan_replicate() local
794 py::class_<Replicated> replicated(pmap_lib, "Replicated"); in BuildPmapSubmodule() local
1580 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
4768 nir_def *replicated = in bifrost_nir_lower_blend_components() local
924 for (bool replicated : replicated_at_leaf_buffers) { in Parameter() local
7940 "replicated": { object