Searched defs:CaptureList (Results 1 – 3 of 3) sorted by relevance
115 struct CaptureList { struct116 CaptureList(size_t capture_size) { in CaptureList() argument123 void captureTensor(const at::Tensor& tensor, bool is_output) { in captureTensor()127 void capture(const IValue& val, bool is_output) { in capture()149 size_t size() const { in size()153 void unpack(Stack& stack, const std::shared_ptr<autograd::Node>& saved_for) { in unpack()180 void release_variables() { in release_variables()187 enum Capture : uint8_t {193 std::vector<Capture> capture_types_;194 std::vector<autograd::SavedVariable> var_captures_;[all …]
505 typedef std::vector<ResolveKey> CaptureList; typedef