Searched defs:replicated (Results 1 – 9 of 9) sorted by relevance
160 nir_ssa_def *replicated[4]; in pan_pack_pure_32() local180 nir_ssa_def *replicated[4]; in pan_pack_pure_16() local252 nir_ssa_def *replicated[4] = { v, v, v, v }; in pan_replicate_4() local
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
794 py::class_<Replicated> replicated(pmap_lib, "Replicated"); in BuildPmapSubmodule() local
1580 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
924 for (bool replicated : replicated_at_leaf_buffers) { in Parameter() local