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