2114ab17979dc3c08b0468c3b799b7836f2d9be7,coremltools/converters/tensorflow/test/test_convnets.py,TFConvNetTest,test_convnet,#TFConvNetTest#,74

Before Change

            x = conv_cell(a, W1)
            W2 = tf.Variable(tf.truncated_normal([3, 3, 4, 2], stddev=0.3))
            x = conv_cell(x, W2)
        self._test_tf_model(graph, {a.op.name: [1, 8, 8, 3]}, [x.op.name])
        self._test_tf_model(graph, {a.op.name: [10, 8, 8, 3]}, [x.op.name])

    def test_convnet_batchnorm(self):
        graph = tf.Graph()

After Change

            W2 = tf.Variable(tf.truncated_normal([3, 3, 4, 2], stddev=0.3))
            x = conv_cell(x, W2)
        self._test_tf_model(graph, {a.op.name: [None, 8, 8, 3]}, [x.op.name],
                batch_sizes=[1,4])

    def test_convnet_batchnorm(self):
        graph = tf.Graph()

Instances