Add support for TensorFlow v2.
This PR provides support for TF 2 and adds a LBNLayer
for keras support.
This PR provides support for TF 2 and adds a LBNLayer
for keras support.
This will create a new commit in order to revert the existing changes.