Skip to content
Permalink
Browse files Browse the repository at this point in the history
Ensure validation sticks in banded_triangular_solve_op
PiperOrigin-RevId: 373275480
Change-Id: Id7717cf275b2d6fdb9441fbbe166d555182d2e79
  • Loading branch information
mihaimaruseac authored and tensorflower-gardener committed May 12, 2021
1 parent 327ef31 commit 0ab2907
Showing 1 changed file with 1 addition and 0 deletions.
Expand Up @@ -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(
Expand Down

0 comments on commit 0ab2907

Please sign in to comment.