| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 323 const int diag_len = std::min(num_rows + std::min(0, diag_index), in ComputeDiagLenAndContentOffset() local
325 const int content_offset = (left_align) ? 0 : (max_diag_len - diag_len); in ComputeDiagLenAndContentOffset()
326 return {diag_len, content_offset}; in ComputeDiagLenAndContentOffset()
359 int diag_len, content_offset; in Compute() local
360 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
415 int diag_len, content_offset; in Compute() local
416 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
421 for (Eigen::Index n = 0; n < diag_len; ++n) { in Compute()
428 const Eigen::Index padding_start = (left_align) ? diag_len : 0; in Compute()
|
| D | matrix_set_diag_op.cc | 227 int diag_len, content_offset; in Compute() local
228 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
234 for (Eigen::Index n = 0; n < diag_len; ++n) { in Compute()
239 for (Eigen::Index n = 0; n < diag_len; ++n) { in Compute()
|
| 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
42 return max_diag_len - diag_len; in ComputeContentOffset()
|
| D | matrix_diag_op_gpu.cu.cc | 40 const int diag_len = min(num_rows + y_offset, num_cols - x_offset); in ComputeContentOffset() local
41 return max_diag_len - diag_len; in ComputeContentOffset()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MatrixDiagPartV3.pbtxt | 72 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
78 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
|
| D | api_def_MatrixSetDiagV3.pbtxt | 75 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
81 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
|
| D | api_def_MatrixDiagV3.pbtxt | 96 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
102 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_diag_ops_test.py | 58 diag_len = min(num_rows + min(0, diag_index), num_cols - max(0, diag_index))
60 padding_len = max_diag_len - diag_len
66 packed_last_dim = (slice(None),) if left_align else (slice(0, diag_len, 1),)
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 128 const int64 diag_len = in GetMatrixDiagonalViaGather() local
130 XlaOp diag_base_indices = BroadcastInDim(Iota(builder, S32, diag_len), in GetMatrixDiagonalViaGather()
131 {diag_len, num_index_dims}, {0}); in GetMatrixDiagonalViaGather()
134 {diag_len}); in GetMatrixDiagonalViaGather()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 413 const int64 diag_len = ComputeDiagLen(diag_index, num_rows, num_cols); in Compile() local
414 const int64 padding_len = max_diag_len - diag_len; in Compile()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 112 diag_len = min(N, M)
115 diag_len -= k
117 diag_len = M - k
120 diag_len += k
122 diag_len = N + k
123 diagonal_ = array_ops.ones([diag_len], dtype=dtype)
|
| /external/eigen/blas/f2c/ |
| D | stbmv.c | 17 uplo_len, ftnlen trans_len, ftnlen diag_len) in stbmv_() argument
|
| D | dtbmv.c | 17 ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in dtbmv_() argument
|
| D | ztbmv.c | 17 *incx, ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ztbmv_() argument
|
| D | ctbmv.c | 17 ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ctbmv_() argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | diag_op_test.py | 63 diag_len = min(num_rows + min(0, diag_index), num_cols - max(0, diag_index))
65 padding_len = max_diag_len - diag_len
71 packed_last_dim = (slice(None),) if left_align else (slice(0, diag_len, 1),)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 7275 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
7281 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
7524 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
7530 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
7835 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
7841 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
|