Searched defs:taus (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.h | 41 XlaOp taus; // Shape: [..., min(m, n)] member
|
| D | qr_expander.cc | 201 auto taus = values[1]; in QrBlock() local
265 auto taus = Zeros( in QrBlock() local
296 XlaOp taus, int64 m, int64 n, PrecisionConfig::Precision precision) { in CompactWYRepresentation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5635 Value taus; in matchAndRewrite() local
5816 void QRBlock(Location loc, Value a, Value *r, Value *taus, Value *vs, in QRBlock()
5835 auto taus = old_values[2]; in QRBlock() local
5994 Value taus, int64_t m, int64_t n, in ComputeWYRepresentation()
6005 const auto taus = old_values[2]; in ComputeWYRepresentation() local
|