xla

80df7368 - try torch func in simple case