swift
[TF] Fix gradients in sum(squeezinAxes:) and mean(squeezinAxes:)
#24164

Merged

[TF] Fix gradients in sum(squeezinAxes:) and mean(squeezinAxes:) #24164

rxwei merged 5 commits into swiftlang:tensorflow from sgugger:patch-1

[TF] Fix gradients in sum(squeezinAxes:) and mean(squeezinAxes:)

3ac59cdd

rxwei requested a review from

rxwei 7 years ago

rxwei requested a review from

dan-zheng 7 years ago

rxwei requested a review from

jekbradbury 7 years ago

dan-zheng approved these changes on 2019-04-19

Remove unused `_vjpMean` function.

fb565e0d

rxwei approved these changes on 2019-04-19

Update Gradients.swift

b2fa44ad

Add test

dfbd9908

Minor edit for consistency.

65752cfa

rxwei merged 9e089ac3 into tensorflow 7 years ago

sgugger deleted the patch-1 branch 7 years ago

Reviewers

rxwei

dan-zheng

jekbradbury

Assignees

No one assigned

Labels

None yet

Milestone

No milestone