Searched refs:rewrite_grad_indexed_slices (Results 1 – 2 of 2) sorted by relevance
31 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, function
302 loop_vars = while_v2_indexed_slices_rewriter.rewrite_grad_indexed_slices(