Searched refs:vtv (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 307 const auto vtv = values[1]; in CompactWYRepresentation() local
310 auto yv = DynamicSliceInMinorDims(vtv, {j}, {1}); in CompactWYRepresentation()
317 return std::vector<XlaOp>{t, vtv}; in CompactWYRepresentation()
326 auto vtv = BatchDot(MaybeConjugate(vs, true), /*transpose_x=*/true, vs, in CompactWYRepresentation() local
328 vtv = Select(TriangleMask(vtv, 0), ZerosLike(vtv), vtv); in CompactWYRepresentation()
329 vtv = (vtv + eye) * tau_scale; in CompactWYRepresentation()
332 ForEachIndex(n, S32, body_fn, {t, vtv}, "wy", builder)); in CompactWYRepresentation()
|