message: Fix values after rename
  Summary: Ref T21349
  Test Plan: HW tests pass
  Reviewers: jakeh, hakons, #tensorflow, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved
  Maniphest Tasks: T21349
diff: @@ -496,8 +496,10 @@ def pipeline(computational_stages,
  if value is None:
  return default
  elif value:
  - return backend_config_pb2.ThreeState.Name(backend_config_pb2.STATE_ON)
  - return backend_config_pb2.ThreeState.Name(backend_config_pb2.STATE_OFF)
  + return backend_config_pb2.ThreeState.Name(
  + backend_config_pb2.THREESTATE_ON)
  + return backend_config_pb2.ThreeState.Name(
  + backend_config_pb2.THREESTATE_OFF)
  offload_weight_update_variables = bool_to_three_state(
  offload_weight_update_variables,
  