Wrap layers that requires two ClassTags into TensorModuleWrapper and move to tf package (#2122)
* add TensorModuleToOperation * fix tests * address comments * fix test * rebase master * fix style
This commit is contained in:
parent
66435fb69e
commit
21dad26cbf