def test_last_conv_layer(self): x = tf.constant(1.0) y = tf.constant(0.0) end_points = { 'silly': y, 'conv2': y, 'conv4': x, 'logits': y, 'conv-1': y, } self.assertEqual(x, networks._last_conv_layer(end_points))