experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bug Fix : resolve cast error by adding cleaner type for MKL

Browse Source
This commit is contained in:
khaled besrour 2020-03-21 18:13:50 +01:00
parent bfafc1acef
commit e5e81a3b52
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/kernels/mkl_conv_ops.h
View File

@ -51,7 +51,7 @@ using mkldnn::stream;
namespace tensorflow {
#ifdef ENABLE_MKLDNN_V1
#define MKLDNN_SIZE_DTYPE long int
#define MKLDNN_SIZE_DTYPE memory::dim
#else
#define MKLDNN_SIZE_DTYPE int
#endif // ENABLE_MKLDNN_V1