Use tf.shape for graph mode support#6

FDecaYed merged 2 commits intoNVIDIA-Merlin:mainNVIDIA-Merlin/distributed-embeddings:mainfrom

edknv:fix_shape_graph_modeedknv/distributed-embeddings:fix_shape_graph_modeCopy head branch name to clipboard