class_docstr: Returns a layer that computes custom logits for BERT squad model.
class_docstr_tok: ["Returns","a","layer","that","computes","custom","logits","for","BERT","squad","model","."]
docstr: Implements build() for the layer.
docstr_tok: ["Implements","build","(",")","for","the","layer","."]
code: def build(self, unused_input_shapes):
      """Implements build() for the layer."""
      self.final_dense = tf.keras.layers.Dense(
          units=2,
          kernel_initializer=self.initializer,
          name="final_dense",
      )
      super(BertSquadLogitsLayer, self).build(unused_input_shapes)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  import tensorflow_hub
  from utils import tf_utils
  from utils.modeling import losses
  from utils.modeling import networks
  from utils.modeling.networks import bert_classifier
  from utils.modeling.networks import bert_pretrainer
  from utils.modeling.networks import bert_span_labeler
label_desc: Use image node_id 2 for calling the BertSquadLogitsLayer obj's underlying member method code with example usage: obj.build(unused_input_shapes) without return types