[AutoDiff] [TF-1288] Supporting differentiable functions with multiple semantic results#38781

Closed

BradLarson wants to merge 12 commits intoswiftlang:mainfrom

BradLarson:TF-1288

Commits on Aug 12, 2022

Initial easing of checks for differentiability of inouts with results.
BradLarson
committed
Initial attempt at allowing for multiple results in getAutoDiffDerivativeFunctionLinearMapType(). Fails for certain non-wrt inouts.
BradLarson
committed
Attempting to detect non-wrt inout parameters, all inouts are now results.
BradLarson
committed
Iterate over all result indices in getAutoDiffPullbackType() and getAutoDiffDifferentialType(), rather than the hardcoded 0 index.
BradLarson
committed
Returning all resultIndices from emitDifferentiabilityWitnessesForFunction(), adding some serialization tests.
BradLarson
committed
Adding test for cross-module registration of functions with multiple semantic results.
BradLarson
committed
Loosening checks that assume only one result. Adjusting tests.
BradLarson
committed
Reworking logic for non-wrt inout parameters. Replacing single result index with all result indices in multiple places.
BradLarson
committed
Consolidated repeated result index generation into a central function.
BradLarson
committed
Converting a last few areas to use multiple result indices.
BradLarson
committed
Added tuple result tests, extracting tuple elements as semantic result types.
BradLarson
committed
Adding @asl's fix for subset parameters thunks involving functions with multiple results, and an activity analysis test representing code that had exposed that issue.

BradLarson
and
asl
committed