diff options
| author |  Tayo Oguntebi <tayo@google.com> | 2017-08-30 13:25:56 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-30 13:30:14 -0700 |
| commit | 1bc5cff9dba8baff9c1476c1708adfeb7898d41a (patch) | |
| tree | 2ac748aaf1276891c1d8a10b93852d8e0777b5d3 /third_party/grpc | |
| parent | e565d1f1fced69789feb10f1ea1241157ec95f93 (diff) | |
[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
Diffstat (limited to 'third_party/grpc')
0 files changed, 0 insertions, 0 deletions