update exports for gradients