Searched defs:AutogradZeroSpecializer (Results 1 – 1 of 1) sorted by relevance
81 struct AutogradZeroSpecializer { struct82 enum class State { Nonzero, Zero, Unknown };84 AutogradZeroSpecializer(std::shared_ptr<Graph> graph) in AutogradZeroSpecializer() function87 void run() { in run()104 bool isBackwardGraph() { in isBackwardGraph()118 void replaceBlockInputsWithGraphInputs(Block* b) { in replaceBlockInputsWithGraphInputs()129 void setStatesOnGraphInputs() { in setStatesOnGraphInputs()152 static void getUsesWithAttribute_( in getUsesWithAttribute_()173 static std::vector<Node*> getUsesWithAttribute(Value* inp, Symbol attr) { in getUsesWithAttribute()179 static Node* getUse(Value* inp, Symbol kind) { in getUse()[all …]