Searched refs:IdentityMatrix (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 323 auto eye = Broadcast(IdentityMatrix(builder, type, n, n), batch_dims); in CompactWYRepresentation()
375 auto q = Broadcast(IdentityMatrix(builder, type, m, m), batch_dims); in BuildQrDecomposition()
382 IdentityMatrix(builder, type, m - i, k), in BuildQrDecomposition()
|
| D | triangular_solve_expander.cc | 105 IdentityMatrix(builder, shape.element_type(), padding, padding); in DiagonalBlocks()
282 IdentityMatrix(builder, shape.element_type(), block_size, block_size); in InvertDiagonalBlocks()
388 a = xla::Add(a, IdentityMatrix(builder, a_shape.element_type(), k, k), in SolveByInvertingDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 33 XlaOp IdentityMatrix(XlaBuilder* builder, PrimitiveType type, int64 m, int64 n);
|
| D | self_adjoint_eig_test.cc | 232 GetAverageAbsoluteError(IdentityMatrix(&builder, F32, 8, 8), in XLA_TEST_F()
|
| D | svd.cc | 265 IdentityMatrix(builder, a_shape.element_type(), m, m), batch_dims); in HouseHolderBidiagonalization()
267 IdentityMatrix(builder, a_shape.element_type(), n, n), batch_dims); in HouseHolderBidiagonalization()
|
| D | self_adjoint_eig.cc | 437 auto v_init = Broadcast(IdentityMatrix(builder, type, m, m), batch_dims); in SelfAdjointEig()
|
| D | matrix.cc | 49 XlaOp IdentityMatrix(XlaBuilder* builder, PrimitiveType type, int64 m, in IdentityMatrix() function
|