xref: /aosp_15_r20/external/pytorch/torch/csrc/lazy/backend/lowering_context.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/csrc/lazy/backend/lowering_context.h>
2 
3 namespace torch {
4 namespace lazy {
5 
LoweringContext(const std::string & name,BackendDevice device)6 LoweringContext::LoweringContext(const std::string& name, BackendDevice device)
7     : device_(std::move(device)) {}
8 
LoweringContext(const std::string & name,BackendDevice device,c10::ArrayRef<const torch::lazy::Node * > post_order,Util::EmissionMap emit_status)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 
GetParametersData() const16 const std::vector<BackendDataPtr>& LoweringContext::GetParametersData() const {
17   return parameters_;
18 }
19 
20 } // namespace lazy
21 } // namespace torch
22