class_docstr: Keras test
class_docstr_tok: ["Keras","test"]
docstr: test_forward_batch_norm
docstr_tok: ["test_forward_batch_norm"]
code: def test_forward_batch_norm(self, keras_mod):
      """test_forward_batch_norm"""
      data = keras_mod.layers.Input(shape=(32, 32, 3))
      batch_norm_funcs = [
          keras_mod.layers.BatchNormalization(
              axis=-1,
              momentum=0.99,
              epsilon=0.001,
              center=True,
              scale=False,
              beta_initializer="zeros",
              gamma_initializer="ones",
              moving_mean_initializer="zeros",
              moving_variance_initializer="ones",
          ),
          keras_mod.layers.BatchNormalization(
              axis=-1,
              momentum=0.99,
              epsilon=0.001,
              center=True,
              scale=True,
              beta_initializer="zeros",
              gamma_initializer="ones",
              moving_mean_initializer="zeros",
              moving_variance_initializer="ones",
          ),
          keras_mod.layers.BatchNormalization(
              axis=-1,
              momentum=0.99,
              epsilon=0.001,
              center=False,
              scale=True,
              beta_initializer="zeros",
              gamma_initializer="ones",
              moving_mean_initializer="zeros",
              moving_variance_initializer="ones",
          ),
          keras_mod.layers.BatchNormalization(
              axis=-1,
              momentum=0.99,
              epsilon=0.001,
              center=False,
              scale=False,
              beta_initializer="zeros",
              gamma_initializer="ones",
              moving_mean_initializer="zeros",
              moving_variance_initializer="ones",
          ),
      ]
      for batch_norm_func in batch_norm_funcs:
          x = batch_norm_func(data)
          keras_model = keras_mod.models.Model(data, x)
          verify_keras_frontend(keras_model)
  
imports: from packaging import version
  import numpy
  from tensorflow import keras
  import keras
  import pytest
  import tvm
  from tvm import relay
  from tvm.contrib import graph_executor
  import tvm.testing
label_desc: Use image node_id 13 for calling the TestKeras obj's underlying member method code with example usage: obj.test_forward_batch_norm(keras_mod) without return types