| 3647 | } |
| 3648 | |
| 3649 | XlaOp Iota(XlaBuilder* builder, PrimitiveType type, int64 size) { |
| 3650 | return builder->Iota(type, size); |
| 3651 | } |
| 3652 | |
| 3653 | XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64 iota_dimension) { |
| 3654 | return builder->Iota(shape, iota_dimension); |