| /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
347 const int content_offset = (left_align) ? 0 : (max_diag_len - diag_len); in ComputeDiagLenAndContentOffset()
348 return {diag_len, content_offset}; in ComputeDiagLenAndContentOffset()
381 int diag_len, content_offset; in Compute() local
382 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
437 int diag_len, content_offset; in Compute() local
438 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
443 for (Eigen::Index n = 0; n < diag_len; ++n) { in Compute()
450 const Eigen::Index padding_start = (left_align) ? diag_len : 0; in Compute()
|
| D | matrix_set_diag_op.cc | 230 int diag_len, content_offset; in Compute() local
231 std::tie(diag_len, content_offset) = ComputeDiagLenAndContentOffset( in Compute()
237 for (Eigen::Index n = 0; n < diag_len; ++n) { in Compute()
242 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 | 55 diag_len = min(num_rows + min(0, diag_index), num_cols - max(0, diag_index))
57 padding_len = max_diag_len - diag_len
63 packed_last_dim = (slice(None),) if left_align else (slice(0, diag_len, 1),)
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 129 const int64_t diag_len = in GetMatrixDiagonalViaGather() local
131 XlaOp diag_base_indices = BroadcastInDim(Iota(builder, S32, diag_len), in GetMatrixDiagonalViaGather()
132 {diag_len, num_index_dims}, {0}); in GetMatrixDiagonalViaGather()
135 {diag_len}); in GetMatrixDiagonalViaGather()
|
| /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
417 const int64_t padding_len = max_diag_len - diag_len; in Compile()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_array_ops.py | 99 diag_len = min(N, M)
102 diag_len -= k
104 diag_len = M - k
107 diag_len += k
109 diag_len = N + k
110 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 | ctbmv.c | 17 ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ctbmv_() argument
|
| D | ztbmv.c | 17 *incx, ftnlen uplo_len, ftnlen trans_len, ftnlen diag_len) in ztbmv_() argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | diag_op_test.py | 59 diag_len = min(num_rows + min(0, diag_index), num_cols - max(0, diag_index))
61 padding_len = max_diag_len - diag_len
67 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 | 8208 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
8214 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
8457 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
8463 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
8768 offset = max_diag_len - diag_len(d) ; if (`align` in {RIGHT_LEFT, RIGHT_RIGHT}
8774 where `diag_len(d) = min(cols - max(d, 0), rows + min(d, 0))`.
|