chlo MLIR Dialect Builder API

Builder Methods

chlo::AcosOp

Creates a new chlo.acos operation.

MlirOp Acos(MlirOp &operand);

chlo::AcoshOp

Creates a new chlo.acosh operation.

MlirOp Acosh(MlirOp &operand);

chlo::AsinAcosKernelOp

Creates a new chlo._asin_acos_kernel operation.

MlirOp AsinAcosKernel(MlirOp &operand);

chlo::AsinOp

Creates a new chlo.asin operation.

MlirOp Asin(MlirOp &operand);

chlo::AsinhOp

Creates a new chlo.asinh operation.

MlirOp Asinh(MlirOp &operand);

chlo::AtanOp

Creates a new chlo.atan operation.

MlirOp Atan(MlirOp &operand);

chlo::AtanhOp

Creates a new chlo.atanh operation.

MlirOp Atanh(MlirOp &operand);

chlo::BesselI1eOp

Creates a new chlo.bessel_i1e operation.

MlirOp BesselI1e(MlirOp &operand);

chlo::BroadcastAddOp

Creates a new chlo.broadcast_add operation.

MlirOp BroadcastAdd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastAndOp

Creates a new chlo.broadcast_and operation.

MlirOp BroadcastAnd(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastAtan2Op

Creates a new chlo.broadcast_atan2 operation.

MlirOp BroadcastAtan2(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastCompareOp

Creates a new chlo.broadcast_compare operation.

MlirOp BroadcastCompare(MlirOp &lhs, MlirOp &rhs, ::mlir::chlo::ComparisonDirection comparison_direction, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {}, /*optional*/::mlir::chlo::ComparisonTypeAttr compare_type = {});

chlo::BroadcastComplexOp

Creates a new chlo.broadcast_complex operation.

MlirOp BroadcastComplex(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastDivOp

Creates a new chlo.broadcast_divide operation.

MlirOp BroadcastDiv(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMaxOp

Creates a new chlo.broadcast_maximum operation.

MlirOp BroadcastMax(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMinOp

Creates a new chlo.broadcast_minimum operation.

MlirOp BroadcastMin(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastMulOp

Creates a new chlo.broadcast_multiply operation.

MlirOp BroadcastMul(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastNextAfterOp

Creates a new chlo.broadcast_next_after operation.

MlirOp BroadcastNextAfter(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastOrOp

Creates a new chlo.broadcast_or operation.

MlirOp BroadcastOr(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastPolygammaOp

Creates a new chlo.broadcast_polygamma operation.

MlirOp BroadcastPolygamma(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastPowOp

Creates a new chlo.broadcast_power operation.

MlirOp BroadcastPow(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastRemOp

Creates a new chlo.broadcast_remainder operation.

MlirOp BroadcastRem(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastSelectOp

Creates a new chlo.broadcast_select operation.

MlirOp BroadcastSelect(MlirOp &pred, MlirOp &on_true, MlirOp &on_false);

chlo::BroadcastShiftLeftOp

Creates a new chlo.broadcast_shift_left operation.

MlirOp BroadcastShiftLeft(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastShiftRightArithmeticOp

Creates a new chlo.broadcast_shift_right_arithmetic operation.

MlirOp BroadcastShiftRightArithmetic(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastShiftRightLogicalOp

Creates a new chlo.broadcast_shift_right_logical operation.

MlirOp BroadcastShiftRightLogical(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastSubOp

Creates a new chlo.broadcast_subtract operation.

MlirOp BroadcastSub(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastXorOp

Creates a new chlo.broadcast_xor operation.

MlirOp BroadcastXor(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::BroadcastZetaOp

Creates a new chlo.broadcast_zeta operation.

MlirOp BroadcastZeta(MlirOp &lhs, MlirOp &rhs, /*optional*/::mlir::DenseI64ArrayAttr broadcast_dimensions = {});

chlo::ConjOp

Creates a new chlo.conj operation.

MlirOp Conj(MlirOp &operand);

chlo::ConstantOp

Creates a new chlo.constant operation.

MlirOp Constant(MlirBuilder &builder, ::mlir::ElementsAttr value);

chlo::CoshOp

Creates a new chlo.cosh operation.

MlirOp Cosh(MlirOp &operand);

chlo::DigammaOp

Creates a new chlo.digamma operation.

MlirOp Digamma(MlirOp &operand);

chlo::ErfInvOp

Creates a new chlo.erf_inv operation.

MlirOp ErfInv(MlirOp &operand);

chlo::ErfOp

Creates a new chlo.erf operation.

MlirOp Erf(MlirOp &operand);

chlo::ErfcOp

Creates a new chlo.erfc operation.

MlirOp Erfc(MlirOp &operand);

chlo::IsInfOp

Creates a new chlo.is_inf operation.

MlirOp IsInf(MlirOp &operand);

chlo::IsNegInfOp

Creates a new chlo.is_neg_inf operation.

MlirOp IsNegInf(MlirOp &operand);

chlo::IsPosInfOp

Creates a new chlo.is_pos_inf operation.

MlirOp IsPosInf(MlirOp &operand);

chlo::LgammaOp

Creates a new chlo.lgamma operation.

MlirOp Lgamma(MlirOp &operand);

chlo::NextAfterOp

Creates a new chlo.next_after operation.

MlirOp NextAfter(MlirOp &x, MlirOp &y);

chlo::PolygammaOp

Creates a new chlo.polygamma operation.

MlirOp Polygamma(MlirOp &n, MlirOp &x);

chlo::RaggedDotOp

Creates a new chlo.ragged_dot operation.

MlirOp RaggedDot(Type resultType, MlirOp &lhs, MlirOp &rhs, MlirOp &group_sizes, ::mlir::chlo::RaggedDotDimensionNumbersAttr ragged_dot_dimension_numbers, /*optional*/::mlir::ArrayAttr precision_config = {});

chlo::SinhOp

Creates a new chlo.sinh operation.

MlirOp Sinh(MlirOp &operand);

chlo::SquareOp

Creates a new chlo.square operation.

MlirOp Square(MlirOp &operand);

chlo::TanOp

Creates a new chlo.tan operation.

MlirOp Tan(MlirOp &operand);

chlo::TopKOp

Creates a new chlo.top_k operation.

SmallVector<MlirOp, 2> TopK(MlirOp &operand, uint64_t k);

chlo::ZetaOp

Creates a new chlo.zeta operation.

MlirOp Zeta(MlirOp &x, MlirOp &q);

Skipped Operations

Unable to generate builder for the following operations: