[TF] Remove unbroadcast(to:) and improve derivative performance.#24408

rxwei wants to merge 2 commits intotensorflowswiftlang/swift:tensorflowfrom

better-unbroadcastingswiftlang/swift:better-unbroadcastingCopy head branch name to clipboard