Searched refs:min_num_rows (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 282 const int64 min_num_rows = in Compile() local
286 OP_REQUIRES(context, num_rows == -1 || num_rows >= min_num_rows, in Compile()
294 num_rows = std::max(min_num_rows, min_num_cols); in Compile()
297 num_rows = min_num_rows; in Compile()
304 OP_REQUIRES(context, num_rows == min_num_rows || num_cols == min_num_cols, in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 221 const int32 min_num_rows = max_diag_len - std::min(upper_diag_index, 0); in Compute() local
223 OP_REQUIRES(context, num_rows == -1 || num_rows >= min_num_rows, in Compute()
231 num_rows = std::max(min_num_rows, min_num_cols); in Compute()
234 num_rows = min_num_rows; in Compute()
238 OP_REQUIRES(context, num_rows == min_num_rows || num_cols == min_num_cols, in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1378 const int32 min_num_rows = max_diag_len - std::min(upper_diag_index, 0); in MatrixDiagV2Shape() local
1381 num_rows = std::max(min_num_rows, min_num_cols); in MatrixDiagV2Shape()
1385 num_rows = min_num_rows; in MatrixDiagV2Shape()
1386 } else if (num_rows < min_num_rows) { in MatrixDiagV2Shape()
1395 if (num_rows != min_num_rows && num_cols != min_num_cols) { in MatrixDiagV2Shape()
1399 "num_rows = ", num_rows, " != min_num_rows = ", min_num_rows, in MatrixDiagV2Shape()
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| D | curses_ui.py | 1130 def _display_lines(self, output, min_num_rows): argument
1153 rows = max(min_num_rows, len(output.lines))
|