Searched refs:grad_output_slices (Results 1 – 1 of 1) sorted by relevance
84 def _rewrite_output_as_tensor(body_grad_graph, grad_output_slices): argument92 new_output = ops.convert_to_tensor_v2(grad_output_slices)94 idx = body_grad_graph.structured_outputs.index(grad_output_slices)100 def _rewrite_input_as_indexed_slices(body_grad_graph, grad_output_slices, argument127 init_slices = _create_grad_indexed_slices_init(grad_output_slices,148 new_output_slices = _rewrite_grad_indexed_slices_output(grad_output_slices,155 grad_output_slices)158 def _create_grad_indexed_slices_init(grad_output_slices, forward_input): argument169 assert isinstance(grad_output_slices, ops.IndexedSlices)171 values_out = grad_output_slices.values[all …]