#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 |