Searched refs:iota_n (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_band_part_op.cc | 66 xla::XlaOp iota_n = xla::Iota(builder, iota_shape, /*iota_dimension=*/1); in Compile() local
68 auto offset = xla::Sub(iota_n, iota_m); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 253 auto iota_n = in QrBlock() local
258 Eq(iota_n, j), in QrBlock()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 404 // iota_n = { M x N matrix with 0,1,...N along the N dimension }
407 // offset = iota_m - iota_n
|
| D | legalize_tf.cc | 5936 auto iota_n = builder->create<IotaOp>( in QRBlock() local
5946 loc, iota_n, j, GetI64ElementsAttr({}, builder), in QRBlock()
|