code: def call(self, x):
      x = tf.reshape(x, (-1, 28, 28, 1))
      x = self.conv1(x)
      x = self.bn1(x)
      x = tf.nn.relu(x)
      x = self.conv2(x)
      x = self.bn2(x)
      x = tf.nn.relu(x)
      x = self.flatten(x)
      x = self.fc(x)
      return x
  
file_path: sagemaker-python-sdk/tests/data/tensorflow_mnist/mnist_v2.py
imports: import argparse
  import numpy
  import os
  import json
  import tensorflow
  from tensorflow.keras.layers import Conv2D, BatchNormalization, Dense, Flatten
label_desc: Use image node_id 2 for calling the LeNet obj's underlying member method code with example usage: obj.call(x) and returns: x