code: def call(self, x, **kwargs):
      x = self.conv1(
          x
      )  # input[batch, 28, 28, 1] output[batch, 26, 26, 32]
      x = self.flatten(x)  # output [batch, 21632]
      x = self.d1(x)  # output [batch, 128]
      return self.d2(x)
  
imports: from tensorflow.keras.layers import Dense, Flatten, Conv2D
  from tensorflow.keras import Model
label_desc: Use image node_id 2 for calling the MyModel obj's underlying member method code with example usage: obj.call(x) and returns: self