Home
last modified time | relevance | path

Searched refs:num_rows1 (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dtridiagonal_solve_op_gpu.cu.cc120 auto num_rows1 = input_matrix_shapes[0].dim_size(1); in ValidateInputMatrixShapes() local
122 OP_REQUIRES(context, num_rows1 == num_rows2, in ValidateInputMatrixShapes()
125 num_rows1, " and ", num_rows2, ".")); in ValidateInputMatrixShapes()