Searched refs:mark_output (Results 1 – 2 of 2) sorted by relevance
484 .mark_output(output_edges[i].input_nr, i); in _compiled_autograd_impl()
76 void mark_output(int input_nr, int output_idx) { in mark_output() function