Searched defs:left_align_superdiagonal (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | matrix_set_diag_op_gpu.cu.cc | 34 const bool left_align_superdiagonal, in ComputeContentOffset()
49 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixSetDiagKernel()
75 const int upper_diag_index, const bool left_align_superdiagonal, in MatrixCopyInputAndSetDiagKernel()
|
| H A D | matrix_diag_op_gpu.cu.cc | 33 const bool left_align_superdiagonal, in ComputeContentOffset()
49 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixDiagKernel()
106 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixDiagPartKernel()
|
| H A D | matrix_diag_op.cc | 331 bool* left_align_superdiagonal, in ReadAlignment()
342 bool left_align_superdiagonal, bool left_align_subdiagonal) { in ComputeDiagLenAndContentOffset()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | matrix_diag_ops.cc | 37 static inline bool IsLeftAligned(int diag_index, bool left_align_superdiagonal, in IsLeftAligned()
45 bool* left_align_superdiagonal, in ReadAlignment()
125 const bool left_align_superdiagonal, in SetMatrixDiag()
|