Searched refs:IdentityMatrix (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.h | 31 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 | qr.cc | 350 auto q = Broadcast(IdentityMatrix(builder, type, m, m), batch_dims); in QRDecomposition()
|
| D | matrix.cc | 41 XlaOp IdentityMatrix(XlaBuilder* builder, PrimitiveType type, int64 m, in IdentityMatrix() function
|
| D | svd.cc | 271 IdentityMatrix(builder, a_shape.element_type(), m, m), batch_dims); in HouseHolderBidiagonalization()
273 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 162 IdentityMatrix(builder, shape.element_type(), block_size, block_size); in InvertDiagonalBlocks()
422 a = xla::Add(a, IdentityMatrix(builder, a_shape.element_type(), k, k), in BuildTriangularSolve()
|