c786d1c50fcc3416e2d684897ddd9501248fc4e1,dataset/models/tf/layers/conv.py,,,#,9

Before Change

    "m": "mip"
}

_LAYERS_KEYS = str(list(C_LAYERS.keys()))
_GROUP_KEYS = _LAYERS_KEYS.replace("v", "p")
C_GROUPS = dict(zip(_LAYERS_KEYS, _GROUP_KEYS))

def _get_layer_fn(fn, dim):
    f = ND_LAYERS[fn]
    return f if callable(f) or f is None else f[dim-1]

After Change

_CONV_LAYERS = {
    : tf.layers.conv1d,
    2: tf.layers.conv2d,
    3: tf.layers.conv3d
}

Instances