Add Tensorflow cumsum, please help review