Searched refs:parameter_replication (Results 1 – 3 of 3) sorted by relevance
270 bool ParseParameterReplication(ParameterReplication* parameter_replication);670 optional<ParameterReplication> parameter_replication; in ParseInstructionRhs() local673 ¶meter_replication}; in ParseInstructionRhs()1725 if (parameter_replication) { in ParseInstructionRhs()1728 parameter_replication->replicated_at_leaf_buffers(); in ParseInstructionRhs()1899 ParameterReplication* parameter_replication) { in ParseParameterReplication() argument1908 parameter_replication->add_replicated_at_leaf_buffers(true); in ParseParameterReplication()1910 parameter_replication->add_replicated_at_leaf_buffers(false); in ParseParameterReplication()2774 ParameterReplication parameter_replication; in ParseAttributeHelper() local2775 if (!ParseParameterReplication(¶meter_replication)) { in ParseAttributeHelper()[all …]
211 xla.ParameterReplication parameter_replication = 61; field
345 if (!proto.parameter_replication().replicated_at_leaf_buffers().empty()) { in CreateFromProto()347 proto.parameter_replication().replicated_at_leaf_buffers()); in CreateFromProto()