Searched defs:diag_len (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 345 const int diag_len = std::min(num_rows + std::min(0, diag_index), in ComputeDiagLenAndContentOffset() local
381 int diag_len, content_offset; in Compute() local
437 int diag_len, content_offset; in Compute() local
|
| D | matrix_set_diag_op.cc | 230 int diag_len, content_offset; in Compute() local
|
| D | matrix_set_diag_op_gpu.cu.cc | 41 const int diag_len = min(num_rows + y_offset, num_cols - x_offset); in ComputeContentOffset() local
|
| D | matrix_diag_op_gpu.cu.cc | 40 const int diag_len = min(num_rows + y_offset, num_cols - x_offset); in ComputeContentOffset() local
|
| /external/eigen/blas/f2c/ |
| D | stbmv.c | 17 uplo_len, ftnlen trans_len, ftnlen diag_len) in stbmv_()
|
| D | dtbmv.c | 17 ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in dtbmv_()
|
| D | ctbmv.c | 17 ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ctbmv_()
|
| D | ztbmv.c | 17 *incx, ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ztbmv_()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 416 const int64_t diag_len = ComputeDiagLen(diag_index, num_rows, num_cols); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 129 const int64_t diag_len = in GetMatrixDiagonalViaGather() local
|