{"record":{"id":"13c3ad3effd4c482","repo":"tensorflow/models","slug":"unknown-output-value-s-output-can-be-eithe-13c3ad","errorCode":null,"errorMessage":"Unknown `output` value \"%s\". `output` can be either \"logits\" or \"predictions\"","messagePattern":"Unknown `output` value \"(.+?)\"\\. `output` can be either \"logits\" or \"predictions\"","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"official/nlp/modeling/layers/mobile_bert_layers.py","lineNumber":472,"sourceCode":"    Args:\n      embedding_table: The embedding table from encoder network.\n      activation: The activation, if any, for the dense layer.\n      initializer: The initializer for the dense layer. Defaults to a Glorot\n        uniform initializer.\n      output: The output style for this layer. Can be either `logits` or\n        `predictions`.\n      output_weights_use_proj: Use projection instead of concating extra output\n        weights, this may reduce the MLM task accuracy but will reduce the model\n        params as well.\n      **kwargs: keyword arguments.\n    \"\"\"\n    super().__init__(**kwargs)\n    self.embedding_table = embedding_table\n    self.activation = activation\n    self.initializer = tf_keras.initializers.get(initializer)\n\n    if output not in ('predictions', 'logits'):\n      raise ValueError(\n          ('Unknown `output` value \"%s\". `output` can be either \"logits\" or '\n           '\"predictions\"') % output)\n    self._output_type = output\n    self._output_weights_use_proj = output_weights_use_proj\n\n  def build(self, input_shape):\n    self._vocab_size, embedding_width = self.embedding_table.shape\n    hidden_size = input_shape[-1]\n    self.dense = tf_keras.layers.Dense(\n        hidden_size,\n        activation=self.activation,\n        kernel_initializer=tf_utils.clone_initializer(self.initializer),\n        name='transform/dense')\n\n    if hidden_size > embedding_width:\n      if self._output_weights_use_proj:\n        self.extra_output_weights = self.add_weight(\n            'output_weights_proj',","sourceCodeStart":454,"sourceCodeEnd":490,"githubUrl":"https://github.com/tensorflow/models/blob/e006f5f0d534913e49c1f1dae87364039fa607e2/official/nlp/modeling/layers/mobile_bert_layers.py#L454-L490","documentation":"Error \"Unknown `output` value \"%s\". `output` can be either \"logits\" or \"predictions\"\" thrown in tensorflow/models.","triggerScenarios":"Thrown at official/nlp/modeling/layers/mobile_bert_layers.py:472 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"e006f5f0d534913e49c1f1dae87364039fa607e2","analyzedAt":"2026-08-24T14:09:15.576Z","schemaVersion":2},"datasetVersion":"2026-08-24T17:17:21.512Z"}