diff options
| author |  Tayo Oguntebi <tayo@google.com> | 2017-11-28 17:20:39 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-28 17:24:07 -0800 |
| commit | e2f9107effb0c5c4cee49a71562865d9e919b3d0 (patch) | |
| tree | aea9ad29a425b94fa0b061b3286ba7d4ba035d12 /tensorflow/compiler/xla/reference_util.h | |
| parent | 73a803fb854fc842700a865d4742ae893ed236d3 (diff) | |
Adds minor-dim pooling tests for cases in which windows exist entirely in
padding. Modifies reference util reduce-window 1D implementation to accept general padding.
PiperOrigin-RevId: 177243527
Diffstat (limited to 'tensorflow/compiler/xla/reference_util.h')
| -rw-r--r-- | tensorflow/compiler/xla/reference_util.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/reference_util.h b/tensorflow/compiler/xla/reference_util.h index 62d455d71a..ee244e9a66 100644 --- a/tensorflow/compiler/xla/reference_util.h +++ b/tensorflow/compiler/xla/reference_util.h @@ -184,6 +184,12 @@ class ReferenceUtil { const std::function<float(float, float)>& reduce_func, const tensorflow::gtl::ArraySlice<int64>& window, const tensorflow::gtl::ArraySlice<int64>& stride, Padding padding); + static std::unique_ptr<std::vector<float>> ReduceWindow1DGeneric( + const tensorflow::gtl::ArraySlice<float>& operand, float init, + const std::function<float(float, float)>& reduce_func, + const tensorflow::gtl::ArraySlice<int64>& window, + const tensorflow::gtl::ArraySlice<int64>& stride, + const tensorflow::gtl::ArraySlice<std::pair<int64, int64>>& padding); static std::unique_ptr<Array4D<float>> ReduceWindow4DGeneric( const Array4D<float>& operand, float init, const std::function<float(float, float)>& reduce_func, |