Searched refs:replica_group_ids (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 51 reserved "replica_group_ids";
|
| D | hlo_parser.cc | 820 optional<std::vector<int64>> replica_group_ids; in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 793 | `replica_group_ids` | `int64` vector | Group ID for each replica. |
800 `replica_group_ids` identifies the group ID of each replica. The group ID must
803 `replica_group_ids` = {0, 1, 2, 3, 0, 1, 2, 3} has eight replicas, there are
806 `replica_group_ids` = {0, 1, 2, 0} for four replicas is invalid.
|