class_docstr: Fixed default name of BatchNormalization to match TpuBatchNormalization.
class_docstr_tok: ["Fixed","default","name","of","BatchNormalization","to","match","TpuBatchNormalization","."]
code: def call(self, inputs, training=None):
      outputs = super().call(inputs, training)
      # A temporary hack for tf. compatibility with keras batch norm.
      for u in self.updates:
          tf1.add_to_collection(tf1.GraphKeys.UPDATE_OPS, u)
      return outputs
  
imports: from typing import Text, Tuple, Union
  import tensorflow
  import argparse
  import tensorflow.compat.v1
  from enum import Enum
  from collections import namedtuple
  from tensorflow.python.eager import tape
  from tensorflow.python.tpu import tpu_function
label_desc: Use image node_id 2 for calling the BatchNormalization obj's underlying member method code with example usage: obj.call(inputs, training) and returns: outputs