| 3655 | } |
| 3656 | |
| 3657 | XlaOp GetDimensionSize(const XlaOp operand, int64 dimension) { |
| 3658 | return operand.builder()->GetDimensionSize(operand, dimension); |
| 3659 | } |
| 3660 | |
| 3661 | XlaOp SetDimensionSize(const XlaOp operand, const XlaOp val, int64 dimension) { |
| 3662 | return operand.builder()->SetDimensionSize(operand, val, dimension); |