Searched refs:min_num_cols (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 284 const int64 min_num_cols = in Compile() local
288 OP_REQUIRES(context, num_cols == -1 || num_cols >= min_num_cols, in Compile()
294 num_rows = std::max(min_num_rows, min_num_cols); in Compile()
299 num_cols = min_num_cols; 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 | 222 const int32 min_num_cols = max_diag_len + std::max(lower_diag_index, 0); in Compute() local
225 OP_REQUIRES(context, num_cols == -1 || num_cols >= min_num_cols, in Compute()
231 num_rows = std::max(min_num_rows, min_num_cols); in Compute()
236 num_cols = min_num_cols; 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 | 1379 const int32 min_num_cols = max_diag_len + std::max(lower_diag_index, 0); in MatrixDiagV2Shape() local
1381 num_rows = std::max(min_num_rows, min_num_cols); in MatrixDiagV2Shape()
1390 num_cols = min_num_cols; in MatrixDiagV2Shape()
1391 } else if (num_cols < min_num_cols) { in MatrixDiagV2Shape()
1395 if (num_rows != min_num_rows && num_cols != min_num_cols) { in MatrixDiagV2Shape()
1400 ", num_cols = ", num_cols, " != min_num_cols = ", min_num_cols); in MatrixDiagV2Shape()
|