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