fix missing

Author: kawakami-masaki0

tests/keras_tests/quantizers_tests/test_activation_inferable_quantizers.py

@@ -241,8 +241,8 @@ def test_uniform_activation_quantizer(self):
         self.assertTrue(quantizer_config['max_range'] == max_range)
 
         # Initialize a random input to quantize between -50 to 50. Input includes positive and negative values.
-        input_tensor = np.random.rand(1, 50, 50, 3) * 50
-        signs = np.where(np.indices((1, 50, 50, 3)).sum(axis=0) % 2 == 0, 1, -1).astype(np.int8)
+        input_tensor = np.random.rand(1, 50, 4, 50) * 50
+        signs = np.where(np.indices((1, 50, 4, 50)).sum(axis=0) % 2 == 0, 1, -1).astype(np.int8)
         input_tensor = tf.constant(input_tensor * signs, dtype=tf.float32)
         fake_quantized_tensor = quantizer(input_tensor)
 
@@ -283,8 +283,8 @@ def test_illegal_range_uniform_activation_quantizer(self):
         # self.assertTrue(quantizer_config['max_range'] == max_range)
 
         # Initialize a random input to quantize between -50 to 50. Input includes positive and negative values.
-        input_tensor = np.random.rand(1, 50, 50, 3) * 50
-        signs = np.where(np.indices((1, 50, 50, 3)).sum(axis=0) % 2 == 0, 1, -1).astype(np.int8)
+        input_tensor = np.random.rand(1, 50, 4, 50) * 50
+        signs = np.where(np.indices((1, 50, 4, 50)).sum(axis=0) % 2 == 0, 1, -1).astype(np.int8)
         input_tensor = tf.constant(input_tensor * signs, dtype=tf.float32)
         fake_quantized_tensor = quantizer(input_tensor)