Searched refs:AddRequiresGradOnOutputNodes (Results 1 – 1 of 1) sorted by relevance
435 void AddRequiresGradOnOutputNodes( in AddRequiresGradOnOutputNodes() function443 AddRequiresGradOnOutputNodes(b, ctx_mapping); in AddRequiresGradOnOutputNodes()456 void AddRequiresGradOnOutputNodes(const std::shared_ptr<Graph>& graph) { in AddRequiresGradOnOutputNodes() function458 AddRequiresGradOnOutputNodes(graph->block(), ctx_mapping); in AddRequiresGradOnOutputNodes()470 AddRequiresGradOnOutputNodes(graph); in CreateAutodiffSubgraphs()