Searched refs:x_after_k (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 98 XlaOp x_after_k = Mul(x, ConvertElementType(Gt(iota, k), type), in House() local
105 auto x_squared = Real(x_after_k * Conj(x_after_k)); in House()
125 auto sigma = Reduce(x_after_k * x_after_k, zero, in House()
147 *v = e_k + Div(x_after_k, divisor, /*broadcast_dimensions=*/batch_dim_ids); in House()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5731 Value x_after_k = builder->create<chlo::BroadcastMulOp>( in House() local
5733 Value x_after_k_sq = builder->create<MulOp>(loc, x_after_k, x_after_k); in House()
5785 StaticBinaryBroadcast<DivOp>(loc, x_after_k, divisor, in House()
|