xla
[XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)).
#6437

Merged

[XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)). #6437

copybara-service merged 1 commit into main from test_574519625

copybara-service force pushed from eee68563 2 years ago

copybara-service changed the title ~~[XLA] Replace Reduce(Broadcast(0), init_val, Sum()) with Broadcast(init_val).~~ [XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)). 2 years ago

copybara-service force pushed 2 years ago

[XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * pr…

7db233cf

copybara-service force pushed to 7db233cf 2 years ago

copybara-service merged 7db233cf into main 2 years ago

copybara-service deleted the test_574519625 branch 2 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

xla [XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)). #6437 Merged

[XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)). #6437

xla
[XLA] Replace Reduce(Broadcast(x), dims, Sum()) with Broadcast(x * prod(dims)).
#6437

Merged