Searched refs:sigma_is_zero (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 101 XlaOp sigma_is_zero; in House() local
114 sigma_is_zero = Eq(sigma, ScalarLike(sigma, 0)); in House()
115 sigma_is_zero = And(sigma_is_zero, Eq(Imag(alpha), ScalarLike(sigma, 0))); in House()
120 *beta = Select(sigma_is_zero, Real(alpha), *beta); in House()
129 sigma_is_zero = Eq(sigma, zero); in House()
133 *beta = Select(sigma_is_zero, alpha, *beta); in House()
136 *tau = Select(sigma_is_zero, ZerosLike(*tau), *tau); in House()
139 Select(sigma_is_zero, Broadcast(ScalarLike(alpha, 1), batch_dims), in House()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5743 Value sigma_is_zero = builder->create<chlo::BroadcastCompareOp>( in House() local
5757 *beta = builder->create<SelectOp>(loc, alpha.getType(), sigma_is_zero, in House()
5763 *tau = builder->create<SelectOp>(loc, alpha.getType(), sigma_is_zero, in House()
5766 divisor = builder->create<SelectOp>(loc, divisor.getType(), sigma_is_zero, in House()
|