Refactor in-memory dataset classes to separate data from task.#147

copybara-service[bot] wants to merge 1 commit intomaintensorflow/gnn:mainfrom

test_480390964tensorflow/gnn:test_480390964Copy head branch name to clipboard