Searched defs:ForwardADLevel (Results 1 – 1 of 1) sorted by relevance
102 struct TORCH_API ForwardADLevel { struct103 ForwardADLevel(uint64_t idx) : idx_(idx) {} in ForwardADLevel() argument111 void erase(const std::shared_ptr<ForwardGrad>& grad) { in erase()136 // - Any of the ForwardADLevel that this ForwardGrad is registered with argument