Commit 1bc5cff9 authored by Tayo Oguntebi's avatar Tayo Oguntebi Committed by TensorFlower Gardener
Browse files

[XLA] Generalize pooling to support N-dimensional inputs. 

Extends the usage of windowed pipeline emitter in pooling to support arbitrary
input dimensions > 3. Users should note that memory usage compounds quickly
with very-high dimensional tensors.

Adds an HLO evaluator test for the R6 pooling case.

PiperOrigin-RevId: 167040226
parent e565d1f1
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment