diff options
| author |  Shanqing Cai <cais@google.com> | 2017-01-23 18:25:25 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-23 18:45:23 -0800 |
| commit | 56fc8834c736878af34f00caa95e7d4a57ab01d2 (patch) | |
| tree | 138c8c7dbb5b97bf0ff24b84c20133569b625553 /tensorflow/core/kernels/xsmm_conv2d.h | |
| parent | 3b4e53b0739804af7e8f51412bac366dd842a3f1 (diff) | |
Merge changes from github.
Change: 145363673
Diffstat (limited to 'tensorflow/core/kernels/xsmm_conv2d.h')
| -rw-r--r-- | tensorflow/core/kernels/xsmm_conv2d.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/xsmm_conv2d.h b/tensorflow/core/kernels/xsmm_conv2d.h index acc50213b0..b439511dc7 100644 --- a/tensorflow/core/kernels/xsmm_conv2d.h +++ b/tensorflow/core/kernels/xsmm_conv2d.h @@ -28,6 +28,11 @@ class OpKernelContext; // XsmmConv2D is a wrapper for libxsmm direct convolutions. +// Returns true if convolution operation specified by function arguments +// can use XsmmConv2D implementation, and false otherwise. +bool CanUseXsmmConv2D(const libxsmm_dnn_conv_desc& desc, + TensorFormat data_format); + namespace functor { template <typename Device, typename T> |