Check checkpoint_options.checkpoint_dir: it may be unfilled in a caller's checkpoint_options.
Check checkpoint_options.checkpoint_dir: it may be unfilled in a caller's checkpoint_options.

PiperOrigin-RevId: 487576430

tensorflow/gnn,tensorflow/gnn