mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a259fbf72c
Follows #134655 Pull Request resolved: https://github.com/pytorch/pytorch/pull/136634 Approved by: https://github.com/Skylion007
20 lines
584 B
C++
20 lines
584 B
C++
#include <torch/csrc/lazy/backend/lowering_context.h>
|
|
|
|
namespace torch::lazy {
|
|
|
|
LoweringContext::LoweringContext(const std::string& name, BackendDevice device)
|
|
: device_(std::move(device)) {}
|
|
|
|
LoweringContext::LoweringContext(
|
|
const std::string& name,
|
|
BackendDevice device,
|
|
c10::ArrayRef<const 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 torch::lazy
|