Home
last modified time | relevance | path

Searched refs:parameter_replication (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc270 bool ParseParameterReplication(ParameterReplication* parameter_replication);
670 optional<ParameterReplication> parameter_replication; in ParseInstructionRhs() local
673 &parameter_replication}; in ParseInstructionRhs()
1725 if (parameter_replication) { in ParseInstructionRhs()
1728 parameter_replication->replicated_at_leaf_buffers(); in ParseInstructionRhs()
1899 ParameterReplication* parameter_replication) { in ParseParameterReplication() argument
1908 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() local
2775 if (!ParseParameterReplication(&parameter_replication)) { in ParseAttributeHelper()
[all …]
Dhlo.proto211 xla.ParameterReplication parameter_replication = 61; field
Dhlo_instruction.cc345 if (!proto.parameter_replication().replicated_at_leaf_buffers().empty()) { in CreateFromProto()
347 proto.parameter_replication().replicated_at_leaf_buffers()); in CreateFromProto()