Home
last modified time | relevance | path

Searched refs:left_align_superdiagonal_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc235 ReadAlignment(context, &left_align_superdiagonal_, in MatrixDiagOp()
320 num_rows, num_cols, left_align_superdiagonal_, in Compile()
325 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixDiagOp
351 ReadAlignment(context, &left_align_superdiagonal_, in MatrixDiagPartOp()
416 if (IsLeftAligned(diag_index, left_align_superdiagonal_, in Compile()
439 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixDiagPartOp
460 ReadAlignment(context, &left_align_superdiagonal_, in MatrixSetDiagOp()
527 num_rows, num_cols, left_align_superdiagonal_, in Compile()
532 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixSetDiagOp
/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op.cc52 functor::ReadAlignment(context, &left_align_superdiagonal_, in MatrixDiagPartOp()
140 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
144 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixDiagPartOp
156 functor::ReadAlignment(context, &left_align_superdiagonal_, in MatrixDiagOp()
259 padding_value, left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
263 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixDiagOp
Dmatrix_set_diag_op.cc49 functor::ReadAlignment(context, &left_align_superdiagonal_, in MatrixSetDiagOp()
166 left_align_superdiagonal_, left_align_subdiagonal_); in Compute()
170 bool left_align_superdiagonal_ = true; member in tensorflow::MatrixSetDiagOp