py_utils.CumSum does not work on TPU for non-float types.
py_utils.CumSum does not work on TPU for non-float types.

PiperOrigin-RevId: 268551933

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo