Searched defs:replicated (Results 1 – 8 of 8) sorted by relevance
292 explicit HloSharding(bool manual, bool replicated, in HloSharding()
104 auto* replicated = proto.add_tuple_shardings(); in TEST_F() local
2620 const auto& replicated = in ParseInstructionRhs() local2733 bool replicated = false; in ParseSingleSharding() local
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
506 const bool replicated = tpu_devices.size() != 1; in BuildParallelExecuteOp() local547 const bool replicated = tpu_devices.size() != 1; in AssignDevicesToReplicatedExecute() local
390 py::class_<Replicated> replicated(pmap_lib, "Replicated"); in BuildPmapSubmodule() local
1694 HloInstruction* replicated, HloInstruction* partition_id, in PerGroupSliceFromReplicated()
858 for (bool replicated : replicated_at_leaf_buffers) { in Parameter() local