code: def test_build(self):
      if backend.config.image_data_format() == "channels_last":
          input_shape = (10, 128, 128, 3)
          output_shape = (32, 10, 126, 126, 64)
      else:
          input_shape = (10, 3, 128, 128)
          output_shape = (32, 10, 64, 126, 126)
      inputs = layers.Input(shape=input_shape, batch_size=32)
      conv_2d_layer = layers.Conv2D(64, (3, 3))
      outputs = layers.TimeDistributed(conv_2d_layer)(inputs)
      self.assertEqual(outputs.shape, output_shape)
  
file_path: keras/keras/layers/rnn/time_distributed_test.py
imports: import numpy
  import pytest
  from keras import backend
  from keras import initializers
  from keras import layers
  from keras import ops
  from keras import testing
label_desc: Use image node_id 2 for calling the TimeDistributedTest obj's underlying member method code with example usage: obj.test_build() without return types