[TF] Reimplement unbroadcast using on-host axis calculation for performance.#24907

rxwei merged 1 commit intoswiftlang:tensorflowswiftlang/swift:tensorflowfrom

rxwei:efficient-unbroadcastrxwei/swift:efficient-unbroadcastCopy head branch name to clipboard