Searched refs:_rewrite_output_as_tensor (Results 1 – 1 of 1) sorted by relevance
79 _rewrite_output_as_tensor(body_grad_graph, output)92 def _rewrite_output_as_tensor(body_grad_graph, grad_output_slices): function