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

rxwei merged 5 commits intoswiftlang:tensorflowswiftlang/swift:tensorflowfrom

sgugger:patch-1sgugger/swift:patch-1Copy head branch name to clipboard