Searched defs:AutogradContext (Results 1 – 1 of 1) sorted by relevance
121 struct TORCH_API AutogradContext { struct124 AutogradContext& operator=(const AutogradContext& other) = delete; argument127 ska::flat_hash_map<std::string, at::IValue> saved_data;163 // The CppNode in the autograd graph that owns this AutogradContext. We need a argument181 AutogradContext ctx_; argument