import tensorflow as tf

            mobilenet_v1.Conv(kernel=[3, 3], stride=2, depth=32),
            mobilenet_v1.DepthSepConv(kernel=[3, 3], stride=1, depth=64),
            mobilenet_v1.DepthSepConv(kernel=[3, 3], stride=2, depth=128),
            mobilenet_v1.DepthSepConv(kernel=[3, 3], stride=1, depth=512)
        ]

        inputs = tf.random_uniform((batch_size, height, width, 3))
        net, end_points = mobilenet_v1.mobilenet_v1_base(
            inputs, final_endpoint='Conv2d_3_pointwise', conv_defs=conv_defs)
        self.assertTrue(net.op.name.startswith('MobilenetV1/Conv2d_3'))
        self.assertListEqual(net.get_shape().as_list(),
                             [batch_size, 56, 56, 512])
