File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ // AUTO-GENERATED by `scripts/generate_torch_ops.py` — DO NOT EDIT.
2+ #ifndef INFINI_OPS_BASE_QR_H_
3+ #define INFINI_OPS_BASE_QR_H_
4+
5+ #include " operator.h"
6+
7+ namespace infini ::ops {
8+
9+ class Qr : public Operator <Qr> {
10+ public:
11+ Qr (const Tensor input, const bool some, Tensor Q, Tensor R)
12+ : input_shape_{input.shape ()},
13+ input_strides_{input.strides ()},
14+ input_type_{input.dtype ()},
15+ Q_shape_{Q.shape ()},
16+ Q_strides_{Q.strides ()},
17+ Q_type_{Q.dtype ()},
18+ R_shape_{R.shape ()},
19+ R_strides_{R.strides ()},
20+ R_type_{R.dtype ()},
21+ some_{some},
22+ device_index_{Q.device ().index ()} {}
23+
24+ virtual void operator ()(const Tensor input, const bool some, Tensor Q, Tensor R) const = 0;
25+
26+ protected:
27+ Tensor::Shape input_shape_;
28+
29+ Tensor::Strides input_strides_;
30+
31+ DataType input_type_;
32+
33+ Tensor::Shape Q_shape_;
34+
35+ Tensor::Strides Q_strides_;
36+
37+ DataType Q_type_;
38+
39+ Tensor::Shape R_shape_;
40+
41+ Tensor::Strides R_strides_;
42+
43+ DataType R_type_;
44+
45+ bool some_{};
46+
47+ int device_index_{0 };
48+ };
49+
50+ } // namespace infini::ops
51+
52+ #endif
You can’t perform that action at this time.
0 commit comments