def augmented_identity(arg):
    identity_op = _array_ops.identity(arg)
    identity_op.op._set_attr(OpHint.FUNCTION_NAME_ATTR, _attr_value_pb2.AttrValue(s=self._function_name))
    identity_op.op._set_attr(OpHint.FUNCTION_UUID_ATTR, _attr_value_pb2.AttrValue(s=self._unique_function_id))
    identity_op.op._set_attr(OpHint.FUNCTION_INPUT_INDEX_ATTR, _attr_value_pb2.AttrValue(i=self._curr_input_index))
    self._curr_input_index += 1
    return identity_op
def augmented_identity(arg):
    identity_op = _array_ops.identity(arg)
    identity_op.op._set_attr(OpHint.FUNCTION_NAME_ATTR, _attr_value_pb2.AttrValue(s=self._function_name))
    identity_op.op._set_attr(OpHint.FUNCTION_UUID_ATTR, _attr_value_pb2.AttrValue(s=self._unique_function_id))
    identity_op.op._set_attr(OpHint.FUNCTION_INPUT_INDEX_ATTR, _attr_value_pb2.AttrValue(i=self._curr_input_index))
    self._curr_input_index += 1
    return identity_op