1 | #include <torch/csrc/lazy/backend/lowering_context.h> |
---|---|
2 | |
3 | namespace torch { |
4 | namespace lazy { |
5 | |
6 | LoweringContext::LoweringContext(const std::string& name, BackendDevice device) |
7 | : device_(std::move(device)) {} |
8 | |
9 | LoweringContext::LoweringContext( |
10 | const std::string& name, |
11 | BackendDevice device, |
12 | c10::ArrayRef<const torch::lazy::Node*> post_order, |
13 | Util::EmissionMap emit_status) |
14 | : device_(std::move(device)), emit_status_(std::move(emit_status)) {} |
15 | |
16 | const std::vector<BackendDataPtr>& LoweringContext::GetParametersData() const { |
17 | return parameters_; |
18 | } |
19 | |
20 | } // namespace lazy |
21 | } // namespace torch |
22 |