ValueError: Input 0 of node gg/a/Assign was passed double from g_small/x:0 incompatible with expected double_ref. Please notice: everything works correctly if the definition of graph_small above is ...