diff options
author | 2018-08-10 00:44:01 -0700 | |
---|---|---|
committer | 2018-08-10 00:48:01 -0700 | |
commit | a769c3acb472ff92e91c1545385c16f04c5bc18a (patch) | |
tree | e901875e2b93bf8bef1f81e585c959722a97673f /tensorflow/core/kernels/constant_op_test.cc | |
parent | 7e3a88be257b0525acc8a35cd0635b6bdfb52eaa (diff) |
In constant op test, do not proceed further if op creation fails.
PiperOrigin-RevId: 208172537
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(); |