1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#include <c10/util/irange.h>
#include <torch/csrc/jit/ir/attributes.h>
#include <torch/csrc/jit/ir/ir.h>
namespace torch {
namespace jit {
AttributeValue::Ptr GraphAttr::clone() const {
return Ptr(new GraphAttr(name, value_->copy()));
}
std::unique_ptr<AttributeValue> GraphsAttr::clone() const {
std::vector<std::shared_ptr<Graph>> copy(value_.size());
for (const auto i : c10::irange(value_.size())) {
copy[i] = value_.at(i)->copy();
}
return Ptr(new GraphsAttr(name, std::move(copy)));
}
} // namespace jit
} // namespace torch
|
