Searched defs:ForwardGrad (Results 1 – 1 of 1) sorted by relevance
127 struct TORCH_API ForwardGrad : std::enable_shared_from_this<ForwardGrad> { struct139 void clear() { in clear()171 // This function removes the tangent for a given level from this ForwardGrad argument186 auto t = (*it).second;193 bool contains(uint64_t level) { in contains()198 bool empty() const { in empty()207 mutable std::mutex mutex_;