Searched defs:left_align_superdiagonal (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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()
|
| 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()
|
| D | matrix_diag_op.cc | 309 bool* left_align_superdiagonal, in ReadAlignment()
320 bool left_align_superdiagonal, bool left_align_subdiagonal) { in ComputeDiagLenAndContentOffset()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 36 static inline bool IsLeftAligned(int diag_index, bool left_align_superdiagonal, in IsLeftAligned()
44 bool* left_align_superdiagonal, in ReadAlignment()
122 const bool left_align_superdiagonal, in SetMatrixDiag()
|