xla

f4f1386e - Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn (#8248)

Fixed tests for basic_gnn_edgecnn and basic_gnn_gcn (#8248) Co-authored-by: Manfei <41607353+ManfeiBai@users.noreply.github.com> Co-authored-by: qihqi <hanq@google.com>