Searched defs:diag_part (Results 1 – 3 of 3) sorted by relevance
458 XlaOp diag_part = xla::Mul(main_diagonal, rhs, broadcasted_dims); in TridiagonalMatMul() local
43 diag_part = array_ops.matrix_diag_part variable
1032 def diag_part(self, name="diag_part"): member in LinearOperator