Merge pull request #49126 from geetachavan1/cherrypicks_X9ZNY

Ensure validation sticks in banded_triangular_solve_op
tensorflow · May 12, 2021 · 86a0d86 · 86a0d86
2 parents 9436ae6 + c03ad1a
commit 86a0d86
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorflow/core/kernels/linalg/banded_triangular_solve_op.cc b/tensorflow/core/kernels/linalg/banded_triangular_solve_op.cc
@@ -217,6 +217,7 @@ class BandedTriangularSolveOpCpu : public OpKernel {
     const Tensor& in1 = ctx->input(1);
 
     ValidateInputTensors(ctx, in0, in1);
+    if (!ctx->status().ok()) return;
 
     MatMulBCast bcast(in0.shape().dim_sizes(), in1.shape().dim_sizes());
     OP_REQUIRES(