Searched defs:replicated (Results 1 – 13 of 13) sorted by relevance
548 const bool replicated = tpu_devices.size() != 1; in AddToParallelExecuteOp() local590 const bool replicated = tpu_devices.size() != 1; in AssignDevicesToReplicatedExecute() 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
419 TestInformation replicated = in runLabPreparersSetup() local486 TestInformation replicated = in runPreparersSetup() local
677 def replicated(mesh: Mesh, rank: int) -> 'Layout': member in Layout
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()
4554 nir_def *replicated = in bifrost_nir_lower_blend_components() local
924 for (bool replicated : replicated_at_leaf_buffers) { in Parameter() local
7940 "replicated": { object
FileMd5$FileChecksum$1.class FileMd5.java package FileMd5$FileChecksum$1 extends com. ...