mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/16232 Differential Revision: D13772974 Pulled By: jamesr66a fbshipit-source-id: b2db370271809e26d3301f8cc98eec567db5e62b
21 lines
510 B
C++
21 lines
510 B
C++
#include <torch/csrc/jit/attributes.h>
|
|
#include <torch/csrc/jit/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 (size_t i = 0; i < value_.size(); ++i) {
|
|
copy[i] = value_.at(i)->copy();
|
|
}
|
|
return Ptr(new GraphsAttr(name, std::move(copy)));
|
|
}
|
|
|
|
|
|
} // namespace jit
|
|
} // namespace torch
|