Partially Codegen Atan2

[ManfeiBai]

Partially codegen atan2

---

Generated LazyIr.h:

class Atan2 : public XlaNode {
 public:
  static torch::lazy::OpKind ClassOpKind() {
    return torch::lazy::OpKind(at::aten::atan2);
  }

  Atan2(const torch::lazy::Value& self, const torch::lazy::Value& other)
      : XlaNode(torch::lazy::OpKind(at::aten::atan2),
              {self, other},
              [&]() { return Atan2OutputShape(self, other); },
              /* num_outputs */ 1,
              torch::lazy::MHash())
  {
    
  }

  std::string ToString() const override {
    std::stringstream ss;
    ss << XlaNode::ToString();
    
    return ss.str();
  }

  

  bool CanBeReused(const torch::lazy::Value& self, const torch::lazy::Value& other) const {
    return false;
    }

  torch_xla::XlaOpVector Lower(LoweringContext* loctx) const override;

  
  

};

[wonjoo-wj]

Thanks, LGTM!

[wonjoo-wj]

Feel free to merge this.

[wonjoo-wj]

Just a note for the future, let's remove "[WIP]" in the PR title before we merge 😄

[ManfeiBai]

Just a note for the future, let's remove "[WIP]" in the PR title before we merge 😄

yeah, sure, thanks, wonjoo, very nice note :P