diff options
Diffstat (limited to 'tensorflow/core/kernels/constant_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/constant_op_test.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/constant_op_test.cc b/tensorflow/core/kernels/constant_op_test.cc index a6baae73d8..0faad11e47 100644 --- a/tensorflow/core/kernels/constant_op_test.cc +++ b/tensorflow/core/kernels/constant_op_test.cc @@ -60,6 +60,7 @@ void ConstantOpTest::PersistentMemoryTrackingTest(bool on_gpu) { std::unique_ptr<OpKernel> op(CreateOpKernel(device_type, device.get(), cpu_allocator(), const_node, TF_GRAPH_DEF_VERSION, &status)); + TF_ASSERT_OK(status); OpKernelContext::Params params; params.device = device.get(); |