Function _bias_add

tensorlayer/layers/normalization.py:100–107

Alternative implementation of tf.nn.bias_add which is compatiable with tensorRT.

(x, b, data_format)

def _bias_add(x, b, data_format):
    """Alternative implementation of tf.nn.bias_add which is compatiable with tensorRT."""
    if data_format == 'NHWC':
        return tf.add(x, b)
    elif data_format == 'NCHW':
        return tf.add(x, b)
    else:
        raise ValueError('invalid data_format: %s' % data_format)


def _compute_shape(tensors):