Searched refs:EinsumDiagonalMask (Results 1 – 1 of 1) sorted by relevance
263 xla::XlaOp EinsumDiagonalMask(XlaOp x, absl::Span<const int64> config) { in EinsumDiagonalMask() function293 return Reduce(EinsumDiagonalMask(x, config), zero, in EinsumDiagonal()321 return EinsumDiagonalMask(x, config); in EinsumInverseDiagonal()