Searched refs:ctx_mapping (Results 1 – 1 of 1) sorted by relevance
363 const ContextMapping& ctx_mapping) { in findRequiresGradForOutput() argument369 if (ctx_mapping.has(use.user) && in findRequiresGradForOutput()370 ctx_mapping.get(use.user) != ctx_mapping.get(diff_graph)) { in findRequiresGradForOutput()388 if (ctx_mapping.has(dg_use.user) && in findRequiresGradForOutput()389 ctx_mapping.get(dg_use.user) != ctx_mapping.get(diff_graph)) { in findRequiresGradForOutput()408 const ContextMapping& ctx_mapping) { in AddRequiresGradToDifferentiableGraph() argument428 diff_graph, diff_graph->output(i), ctx_mapping); in AddRequiresGradToDifferentiableGraph()437 const ContextMapping& ctx_mapping) { in AddRequiresGradOnOutputNodes() argument440 AddRequiresGradToDifferentiableGraph(n, ctx_mapping); in AddRequiresGradOnOutputNodes()443 AddRequiresGradOnOutputNodes(b, ctx_mapping); in AddRequiresGradOnOutputNodes()[all …]