Searched defs:taus (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | qr_expander.cc | 225 auto taus = values[1]; in QrBlock() local
289 auto taus = Zeros( in QrBlock() local
320 XlaOp taus, int64_t m, int64_t n, PrecisionConfig::Precision precision) { in CompactWYRepresentation()
401 auto taus = Zeros(builder, ShapeUtil::MakeShape(type, taus_dims)); in BuildQrDecomposition() local
437 XlaOp a, XlaOp taus, int64_t block_size, in ProductOfElementaryHouseholderReflectors()
545 XlaOp taus = in ExpandInstruction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | qr.cc | 58 auto taus = GetTupleElement(qr, 1); in Qr() local
69 XlaOp ProductOfElementaryHouseholderReflectors(XlaOp a, XlaOp taus) { in ProductOfElementaryHouseholderReflectors()
|
| H A D | qr.h | 36 XlaOp taus; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 6554 Value taus; in matchAndRewrite() local
6730 void QRBlock(Location loc, Value a, Value *r, Value *taus, Value *vs, in QRBlock()
6749 auto taus = old_values[2]; in QRBlock() local
6906 Value taus, int64_t m, int64_t n, in ComputeWYRepresentation()
6917 const auto taus = old_values[2]; in ComputeWYRepresentation() local
|