diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc b/tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc index 6381b527a1..dc6b88e953 100644 --- a/tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc +++ b/tensorflow/core/kernels/mkl_conv_grad_filter_ops.cc @@ -37,9 +37,9 @@ limitations under the License. #include "tensorflow/core/util/use_cudnn.h" #include "tensorflow/core/util/work_sharder.h" +#include "tensorflow/core/util/mkl_util.h" #include "third_party/mkl/include/mkl_dnn.h" #include "third_party/mkl/include/mkl_dnn_types.h" -#include "tensorflow/core/util/mkl_util.h" namespace tensorflow { @@ -266,8 +266,11 @@ class MklConv2DCustomBackpropFilterOp : public OpKernel { int input_offsets[2]; size_t conv_strides[2]; MklShape input_shape, grad_filter_shape, out_backprop_shape; - dnnPrimitive_t prim_conv_bwdfilter, convert_bwdfilter; - dnnLayout_t lt_input, lt_grad_filter, lt_out_backprop; + dnnPrimitive_t prim_conv_bwdfilter = nullptr; + dnnPrimitive_t convert_bwdfilter = nullptr; + dnnLayout_t lt_input = nullptr; + dnnLayout_t lt_grad_filter = nullptr; + dnnLayout_t lt_out_backprop = nullptr; void* conv_res[dnnResourceNumber]; void MklCleanup() { @@ -409,7 +412,7 @@ class MklConv2DCustomBackpropFilterOp : public OpKernel { }; #define REGISTER_MKL_FILTER_KERNELS(T) \ - REGISTER_KERNEL_BUILDER(Name("_MklConv2DBackpropFilter") \ + REGISTER_KERNEL_BUILDER(Name("_MklConv2DBackpropFilter") \ .Device(DEVICE_CPU) \ .TypeConstraint<T>("T") \ .Label(mkl_op_registry::kMklOpLabel), \ |