def _conv2d()

in benchmarks/horovod-resnet/train_imagenet_resnet_hvd.py [0:0]

• 14 lines of code
• 4 McCabe index (conditional complexity)

    def _conv2d(self, inputs, activation, *args, **kwargs):
        x = tf.layers.conv2d(
            inputs,
            data_format=self.data_format,
            use_bias=not self.use_batch_norm,
            kernel_initializer=self.conv_initializer,
            activation=None if self.use_batch_norm else activation,
            *args,
            **kwargs
        )
        if self.use_batch_norm:
            x = self.batch_norm(x)
            x = activation(x) if activation is not None else x
        return x