Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_set_diag_op_gpu.cu.cc34 const bool left_align_superdiagonal, in ComputeContentOffset() argument
36 const bool left_align = (diag_index >= 0 && left_align_superdiagonal) || 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()
75 const int upper_diag_index, const bool left_align_superdiagonal, in MatrixCopyInputAndSetDiagKernel() argument
88 left_align_superdiagonal, left_align_subdiagonal); in MatrixCopyInputAndSetDiagKernel()
108 const bool left_align_superdiagonal, in Compute()
123 upper_diag_index, left_align_superdiagonal, left_align_subdiagonal, in Compute()
131 lower_diag_index, upper_diag_index, left_align_superdiagonal, in Compute()
Dmatrix_diag_op_gpu.cu.cc33 const bool left_align_superdiagonal, in ComputeContentOffset() argument
35 const bool left_align = (diag_index >= 0 && left_align_superdiagonal) || 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()
80 const bool left_align_superdiagonal, 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()
141 const bool left_align_superdiagonal, in Compute()
157 padding_value, left_align_superdiagonal, left_align_subdiagonal, in Compute()
Dmatrix_diag_op.h32 bool* left_align_superdiagonal,
45 bool left_align_superdiagonal, bool left_align_subdiagonal);
54 const T padding_value, const bool left_align_superdiagonal,
66 const bool left_align_superdiagonal, const bool left_align_subdiagonal);
Dmatrix_diag_op.cc309 bool* left_align_superdiagonal, in ReadAlignment() argument
314 *left_align_superdiagonal = align == "LEFT_LEFT" || align == "LEFT_RIGHT"; in ReadAlignment()
320 bool left_align_superdiagonal, bool left_align_subdiagonal) { in ComputeDiagLenAndContentOffset() argument
321 const bool left_align = (diag_index >= 0 && left_align_superdiagonal) || in ComputeDiagLenAndContentOffset()
337 const bool left_align_superdiagonal, in Compute()
348 &padding_value, &left_align_superdiagonal, in Compute()
362 left_align_superdiagonal, left_align_subdiagonal); in Compute()
393 const bool left_align_superdiagonal, 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.cc211 const bool left_align_superdiagonal, in Compute()
218 &num_diags, &left_align_superdiagonal, 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.h35 const bool left_align_superdiagonal,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc36 static inline bool IsLeftAligned(int diag_index, bool left_align_superdiagonal, in IsLeftAligned() argument
38 return (diag_index >= 0 && left_align_superdiagonal) || in IsLeftAligned()
44 bool* left_align_superdiagonal, in ReadAlignment() argument
49 *left_align_superdiagonal = align == "LEFT_LEFT" || align == "LEFT_RIGHT"; in ReadAlignment()
122 const bool left_align_superdiagonal, in SetMatrixDiag() argument
196 if ((diag_index > 0 && left_align_superdiagonal) || in SetMatrixDiag()
206 if (IsLeftAligned(diag_index, left_align_superdiagonal, in SetMatrixDiag()