Searched refs:v_broadcast (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 217 auto v_broadcast = Reshape(v, shape); in QrBlock() local
222 auto vva = BatchDot(MaybeConjugate(v_broadcast, true), in QrBlock()
224 vva = BatchDot(v_broadcast, true, vva, false, precision); in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5851 auto v_broadcast = builder->create<ReshapeOp>( in QRBlock() local
5857 auto vva = BatchDot(loc, v_broadcast, false, a, false, num_batch_dims, in QRBlock()
5859 vva = BatchDot(loc, v_broadcast, true, vva, false, num_batch_dims, in QRBlock()
|