Searched refs:taus (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 201 auto taus = values[1]; in QrBlock() local
256 auto taus_zeros = ZerosLike(taus); in QrBlock()
261 taus = taus + taus_update; in QrBlock()
262 return std::vector<XlaOp>{a, taus}; in QrBlock()
265 auto taus = Zeros( in QrBlock() local
270 {a, taus}, "qr", builder)); in QrBlock()
274 result.taus = values[1]; in QrBlock()
296 XlaOp taus, int64 m, int64 n, PrecisionConfig::Precision precision) { in CompactWYRepresentation() argument
320 auto tau_scale = BroadcastInDim(-taus, ConcatVectors(batch_dims, {1, n}), in CompactWYRepresentation()
391 auto t, CompactWYRepresentation(type, batch_dims, y, qr_block.taus, in BuildQrDecomposition()
|
| D | qr_expander.h | 41 XlaOp taus; // Shape: [..., min(m, n)] member
49 XlaOp taus, int64 m, int64 n, PrecisionConfig::Precision precision);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5635 Value taus; in matchAndRewrite() local
5637 QRBlock(op.getLoc(), a_block, &r_block, &taus, &vs, &rewriter); in matchAndRewrite()
5644 batch_dims, vs, taus, m - i, k, &rewriter); in matchAndRewrite()
5816 void QRBlock(Location loc, Value a, Value *r, Value *taus, Value *vs, in QRBlock() argument
5835 auto taus = old_values[2]; in QRBlock() local
5942 loc, taus.getType(), taus_zeros, in QRBlock()
5943 GetI64ElementsAttr(taus.getType().cast<RankedTensorType>().getShape(), in QRBlock()
5949 loc, taus.getType(), taus_mask, in QRBlock()
5954 taus = builder->create<AddOp>(loc, taus, taus_update); in QRBlock()
5955 new_values->assign({a, vs, taus}); in QRBlock()
[all …]
|
| /external/hyphenation-patterns/de/ |
| D | hyph-de-ch-1901.pat.txt | 19574 1taus
|
| D | hyph-de-1901.pat.txt | 19762 1taus
|
| /external/rust/crates/csv/examples/data/bench/ |
| D | worldcitiespop.csv | 4364 pk,taus kili,Taus Kili,02,,30.148723,68.621112
|
| /external/kotlinx.coroutines/benchmarks/src/jmh/resources/ |
| D | ospd.txt.gz | |