Home
last modified time | relevance | path

Searched refs:mapped_diag_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op_gpu.cu.cc113 const int mapped_diag_index = in MatrixDiagPartKernel() local
115 const int diag_index = upper_diag_index - mapped_diag_index; in MatrixDiagPartKernel()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc168 const int64 mapped_diag_index = upper_diag_index - diag_index; in SetMatrixDiag() local
170 xla::SliceInDim(diag, mapped_diag_index, mapped_diag_index + 1, 1, in SetMatrixDiag()