Home
last modified time | relevance | path

Searched refs:left_align_subdiagonal (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_set_diag_op_gpu.cu.cc35 const bool left_align_subdiagonal) { in ComputeContentOffset() argument
37 (diag_index <= 0 && left_align_subdiagonal); in ComputeContentOffset()
49 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixSetDiagKernel() argument
59 left_align_superdiagonal, left_align_subdiagonal); in MatrixSetDiagKernel()
76 const bool left_align_subdiagonal, const Scalar* __restrict__ input_ptr, in MatrixCopyInputAndSetDiagKernel() argument
88 left_align_superdiagonal, left_align_subdiagonal); in MatrixCopyInputAndSetDiagKernel()
109 const bool left_align_subdiagonal) { in Compute()
123 upper_diag_index, left_align_superdiagonal, left_align_subdiagonal, in Compute()
132 left_align_subdiagonal, input.data(), diag.data(), output.data())); in Compute()
Dmatrix_diag_op_gpu.cu.cc34 const bool left_align_subdiagonal) { in ComputeContentOffset() argument
36 (diag_index <= 0 && left_align_subdiagonal); in ComputeContentOffset()
49 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixDiagKernel() argument
60 left_align_superdiagonal, left_align_subdiagonal); in MatrixDiagKernel()
81 const bool left_align_subdiagonal) { in Compute()
96 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
106 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixDiagPartKernel() argument
118 left_align_superdiagonal, left_align_subdiagonal); in MatrixDiagPartKernel()
142 const bool left_align_subdiagonal) { in Compute()
157 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
Dmatrix_diag_op.h33 bool* left_align_subdiagonal);
45 bool left_align_superdiagonal, bool left_align_subdiagonal);
55 const bool left_align_subdiagonal);
66 const bool left_align_superdiagonal, const bool left_align_subdiagonal);
Dmatrix_diag_op.cc310 bool* left_align_subdiagonal) { in ReadAlignment() argument
315 *left_align_subdiagonal = align == "LEFT_LEFT" || align == "RIGHT_LEFT"; in ReadAlignment()
320 bool left_align_superdiagonal, bool left_align_subdiagonal) { in ComputeDiagLenAndContentOffset() argument
322 (diag_index <= 0 && left_align_subdiagonal); in ComputeDiagLenAndContentOffset()
338 const bool left_align_subdiagonal) { in Compute()
349 &left_align_subdiagonal](Eigen::Index begin, in Compute()
362 left_align_superdiagonal, left_align_subdiagonal); in Compute()
394 const bool left_align_subdiagonal) { in Compute()
407 &left_align_superdiagonal, &left_align_subdiagonal]( in Compute()
418 left_align_superdiagonal, left_align_subdiagonal); in Compute()
[all …]
Dmatrix_set_diag_op.cc212 const bool left_align_subdiagonal) { in Compute()
219 &left_align_subdiagonal](Eigen::Index begin, in Compute()
230 left_align_superdiagonal, left_align_subdiagonal); in Compute()
272 const bool left_align_superdiagonal, const bool left_align_subdiagonal); \
Dmatrix_set_diag_op.h36 const bool left_align_subdiagonal);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc37 bool left_align_subdiagonal) { in IsLeftAligned() argument
39 (diag_index <= 0 && left_align_subdiagonal); in IsLeftAligned()
45 bool* left_align_subdiagonal) { in ReadAlignment() argument
50 *left_align_subdiagonal = align == "LEFT_LEFT" || align == "RIGHT_LEFT"; in ReadAlignment()
123 const bool left_align_subdiagonal) { in SetMatrixDiag() argument
197 (diag_index < 0 && !left_align_subdiagonal)) { in SetMatrixDiag()
207 left_align_subdiagonal)) { in SetMatrixDiag()