Searched refs:while_op_output (Results 1 – 1 of 1) sorted by relevance
554 def _preprocess_grad(grad, body_graph_output, while_op_input, while_op_output): argument587 if (while_op_output.dtype in (dtypes.resource, dtypes.variant) and589 return _zeros_like(while_op_input, while_op_output)