Searched refs:iota_mn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 210 auto iota_mn = Iota( in QrBlock() local
223 Select(Lt(j, iota_mn), a, ZerosLike(a)), precision); in QrBlock()
247 a = Select(Eq(iota_mn, j), new_x, a); in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5904 auto iota_mn = builder->create<IotaOp>( in QRBlock() local
5909 loc, iota_mn, j, GetI64ElementsAttr({}, builder), in QRBlock()
6019 auto iota_mn = builder->create<IotaOp>( in ComputeWYRepresentation() local
6031 loc, iota_mn, j, GetI64ElementsAttr({}, builder), in ComputeWYRepresentation()
|