Searched refs:tau_scale (Results 1 – 1 of 1) sorted by relevance
320 auto tau_scale = BroadcastInDim(-taus, ConcatVectors(batch_dims, {1, n}), in CompactWYRepresentation() local329 vtv = (vtv + eye) * tau_scale; in CompactWYRepresentation()