Searched refs:MaybeTransposeInMinorDims (Results 1 – 5 of 5) sorted by relevance
42 xla::XlaOp input = xla::MaybeTransposeInMinorDims(ctx->Input(0), adjoint_); in Compile()
246 XlaOp l = BuildCholesky(MaybeTransposeInMinorDims(a, !options.lower()), in ExpandInstruction()249 MaybeTransposeInMinorDims(l, !options.lower()); in ExpandInstruction()
139 xla::XlaOp MaybeTransposeInMinorDims(xla::XlaOp x, bool transpose);
710 XlaOp MaybeTransposeInMinorDims(XlaOp x, bool transpose) { in MaybeTransposeInMinorDims() function
484 a_tri = MaybeTransposeInMinorDims( in XLA_TEST_P()