yapf

          in_layers=[in_layer, self.degree_slice, self.membership] +

          self.deg_adjs)

      batch_norm1 = BatchNorm(in_layers=[gc1])

            in_layer = GraphPool(in_layers=[batch_norm1, self.degree_slice, self.membership]

                                           + self.deg_adjs)

        dense = Dense(out_channels=self.dense_layer_size, activation_fn=tf.nn.relu, in_layers=[in_layer])

      in_layer = GraphPool(

          in_layers=[batch_norm1, self.degree_slice, self.membership

                    ] + self.deg_adjs)

    dense = Dense(

        out_channels=self.dense_layer_size,

        activation_fn=tf.nn.relu,

        in_layers=[in_layer])

    dense = Dropout(self.dropout, in_layers=[dense])

    batch_norm3 = BatchNorm(in_layers=[dense])

    readout = GraphGather(