tensorflow/core/ops/compat/ops_history_v2/CollectiveAssignGroupV2.pbtxt

op {
  name: "CollectiveAssignGroupV2"
  input_arg {
    name: "group_assignment"
    type: DT_INT32
  }
  input_arg {
    name: "device_index"
    type: DT_INT32
  }
  output_arg {
    name: "group_key"
    type: DT_INT32
  }
  is_stateful: true
  is_distributed_communication: true
}