Home
last modified time | relevance | path

Searched defs:left_align_subdiagonal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_set_diag_op_gpu.cu.cc35 const bool left_align_subdiagonal) { in ComputeContentOffset()
49 const bool left_align_superdiagonal, const bool left_align_subdiagonal, in MatrixSetDiagKernel()
76 const bool left_align_subdiagonal, const Scalar* __restrict__ input_ptr, in MatrixCopyInputAndSetDiagKernel()
Dmatrix_diag_op_gpu.cu.cc34 const bool left_align_subdiagonal) { 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()
Dmatrix_diag_op.cc310 bool* left_align_subdiagonal) { in ReadAlignment()
320 bool left_align_superdiagonal, bool left_align_subdiagonal) { in ComputeDiagLenAndContentOffset()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc37 bool left_align_subdiagonal) { in IsLeftAligned()
45 bool* left_align_subdiagonal) { in ReadAlignment()
123 const bool left_align_subdiagonal) { in SetMatrixDiag()