xla

e2aedb78 - support amp (auto mixed precision) (#2654)

support amp (auto mixed precision) (#2654) * support amp (auto mixed precision) * update amp to match the code change in pytorch * add amp op cpp test * xla amp could handle shape (1, ) now. Co-authored-by: Chengji Yao <chengji.yao@bytedance.com>