diff options
| author |  Guozhong Zhuang <guozhong.zhuang@intel.com> | 2018-08-20 15:06:18 -0700 |
|---|---|---|
| committer | Guozhong Zhuang <guozhong.zhuang@intel.com> | 2018-08-20 15:06:18 -0700 |
| commit | 86d9ce130c5691cdba16024f7cc7987082acd294 (patch) | |
| tree | ff3bb2ad111fdeb1afcdd739b4f4b41c64ee6966 /tensorflow/core/util | |
| parent | 2459fd5f01b2a135335b588803fd8946ea761387 (diff) | |
do not control primitive caching for small batch size
Diffstat (limited to 'tensorflow/core/util')
| -rw-r--r-- | tensorflow/core/util/mkl_util.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/util/mkl_util.h b/tensorflow/core/util/mkl_util.h index ee02debddf..5d28b8a3e5 100644 --- a/tensorflow/core/util/mkl_util.h +++ b/tensorflow/core/util/mkl_util.h @@ -105,6 +105,8 @@ typedef enum { Dim3d_I = 1 } MklDnnDims3D; +static const int kSmallBatchSize = 32; + #ifdef INTEL_MKL_ML_ONLY class MklShape { public: |