diff options
Diffstat (limited to 'tensorflow/core/kernels/mkl_concat_op.cc')
-rw-r--r-- | tensorflow/core/kernels/mkl_concat_op.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/mkl_concat_op.cc b/tensorflow/core/kernels/mkl_concat_op.cc index 27930c44a6..094ab1c6c6 100644 --- a/tensorflow/core/kernels/mkl_concat_op.cc +++ b/tensorflow/core/kernels/mkl_concat_op.cc @@ -265,6 +265,7 @@ class MklConcatOp : public OpKernel { s.GetDimension() > 0 ? s.GetSizes()[concat_dim] : 1; } mkl_context.MklCreateInputLayouts(context, input_shapes); + OP_REQUIRES_OK(context, context->status()); CHECK_EQ(dnnConcatCreate_F32(&mkl_context.prim_concat, NULL, N, &mkl_context.lt_inputs[0]), @@ -316,12 +317,14 @@ class MklConcatOp : public OpKernel { mkl_context.mkl_tmp_tensors.resize(N); mkl_context.MklPrepareConcatInputs(context, input_tensors); + OP_REQUIRES_OK(context, context->status()); // Execute primitive. CHECK_EQ(dnnExecute_F32(mkl_context.prim_concat, mkl_context.concat_res), E_SUCCESS); mkl_context.MklCleanup(); + OP_REQUIRES_OK(context, context->status()); } private: |