xla

4e5c987e - fix a bug in tensor.flatten