Searched refs:CheckSecondToLastDimension (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 37 Status CheckSecondToLastDimension(const Shape& op_shape, int64 rank, in CheckSecondToLastDimension() function
101 TF_RETURN_IF_ERROR(CheckSecondToLastDimension(lower_diagonal_shape, rank, 1, in CheckSystemAndReturnNumEquations()
104 CheckSecondToLastDimension(main_diagonal_shape, rank, 1, "diagonal")); in CheckSystemAndReturnNumEquations()
105 TF_RETURN_IF_ERROR(CheckSecondToLastDimension(upper_diagonal_shape, rank, 1, in CheckSystemAndReturnNumEquations()
|